agora inbox for [email protected]
help / color / mirror / Atom feed[PATCH v30 01/11] Add a syntax to create Incrementally Maintainable Materialized Views
496+ messages / 2 participants
[nested] [flat]
* [PATCH v30 01/11] Add a syntax to create Incrementally Maintainable Materialized Views
@ 2019-12-20 01:05 Yugo Nagata <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Yugo Nagata @ 2019-12-20 01:05 UTC (permalink / raw)
Allow to create Incrementally Maintainable Materialized View (IMMV)
by using INCREMENTAL option in CREATE MATERIALIZED VIEW command
as follow:
CREATE [INCREMANTAL] MATERIALIZED VIEW xxxxx AS SELECT ....;
---
src/backend/parser/gram.y | 32 +++++++++++++++++++++-----------
src/include/nodes/primnodes.h | 1 +
src/include/parser/kwlist.h | 1 +
3 files changed, 23 insertions(+), 11 deletions(-)
diff --git a/src/backend/parser/gram.y b/src/backend/parser/gram.y
index 130f7fc7c3..7ba972ae5d 100644
--- a/src/backend/parser/gram.y
+++ b/src/backend/parser/gram.y
@@ -466,6 +466,7 @@ static Node *makeRecursiveViewSelect(char *relname, List *aliases, Node *query);
%type <range> OptTempTableName
%type <into> into_clause create_as_target create_mv_target
+%type <boolean> incremental
%type <defelt> createfunc_opt_item common_func_opt_item dostmt_opt_item
%type <fun_param> func_arg func_arg_with_default table_func_column aggr_arg
@@ -719,7 +720,7 @@ static Node *makeRecursiveViewSelect(char *relname, List *aliases, Node *query);
HANDLER HAVING HEADER_P HOLD HOUR_P
IDENTITY_P IF_P ILIKE IMMEDIATE IMMUTABLE IMPLICIT_P IMPORT_P IN_P INCLUDE
- INCLUDING INCREMENT INDENT INDEX INDEXES INHERIT INHERITS INITIALLY INLINE_P
+ INCLUDING INCREMENT INCREMENTAL INDENT INDEX INDEXES INHERIT INHERITS INITIALLY INLINE_P
INNER_P INOUT INPUT_P INSENSITIVE INSERT INSTEAD INT_P INTEGER
INTERSECT INTERVAL INTO INVOKER IS ISNULL ISOLATION
@@ -4696,32 +4697,34 @@ opt_with_data:
*****************************************************************************/
CreateMatViewStmt:
- CREATE OptNoLog MATERIALIZED VIEW create_mv_target AS SelectStmt opt_with_data
+ CREATE OptNoLog incremental MATERIALIZED VIEW create_mv_target AS SelectStmt opt_with_data
{
CreateTableAsStmt *ctas = makeNode(CreateTableAsStmt);
- ctas->query = $7;
- ctas->into = $5;
+ ctas->query = $8;
+ ctas->into = $6;
ctas->objtype = OBJECT_MATVIEW;
ctas->is_select_into = false;
ctas->if_not_exists = false;
/* cram additional flags into the IntoClause */
- $5->rel->relpersistence = $2;
- $5->skipData = !($8);
+ $6->rel->relpersistence = $2;
+ $6->skipData = !($9);
+ $6->ivm = $3;
$$ = (Node *) ctas;
}
- | CREATE OptNoLog MATERIALIZED VIEW IF_P NOT EXISTS create_mv_target AS SelectStmt opt_with_data
+ | CREATE OptNoLog incremental MATERIALIZED VIEW IF_P NOT EXISTS create_mv_target AS SelectStmt opt_with_data
{
CreateTableAsStmt *ctas = makeNode(CreateTableAsStmt);
- ctas->query = $10;
- ctas->into = $8;
+ ctas->query = $11;
+ ctas->into = $9;
ctas->objtype = OBJECT_MATVIEW;
ctas->is_select_into = false;
ctas->if_not_exists = true;
/* cram additional flags into the IntoClause */
- $8->rel->relpersistence = $2;
- $8->skipData = !($11);
+ $9->rel->relpersistence = $2;
+ $9->skipData = !($12);
+ $9->ivm = $3;
$$ = (Node *) ctas;
}
;
@@ -4738,9 +4741,14 @@ create_mv_target:
$$->tableSpaceName = $5;
$$->viewQuery = NULL; /* filled at analysis time */
$$->skipData = false; /* might get changed later */
+ $$->ivm = false;
}
;
+incremental: INCREMENTAL { $$ = true; }
+ | /*EMPTY*/ { $$ = false; }
+ ;
+
OptNoLog: UNLOGGED { $$ = RELPERSISTENCE_UNLOGGED; }
| /*EMPTY*/ { $$ = RELPERSISTENCE_PERMANENT; }
;
@@ -17252,6 +17260,7 @@ unreserved_keyword:
| INCLUDE
| INCLUDING
| INCREMENT
+ | INCREMENTAL
| INDENT
| INDEX
| INDEXES
@@ -17820,6 +17829,7 @@ bare_label_keyword:
| INCLUDE
| INCLUDING
| INCREMENT
+ | INCREMENTAL
| INDENT
| INDEX
| INDEXES
diff --git a/src/include/nodes/primnodes.h b/src/include/nodes/primnodes.h
index 4a154606d2..5d61964f37 100644
--- a/src/include/nodes/primnodes.h
+++ b/src/include/nodes/primnodes.h
@@ -154,6 +154,7 @@ typedef struct IntoClause
/* materialized view's SELECT query */
Node *viewQuery pg_node_attr(query_jumble_ignore);
bool skipData; /* true for WITH NO DATA */
+ bool ivm; /* true for WITH IVM */
} IntoClause;
diff --git a/src/include/parser/kwlist.h b/src/include/parser/kwlist.h
index 2331acac09..ab92b54642 100644
--- a/src/include/parser/kwlist.h
+++ b/src/include/parser/kwlist.h
@@ -207,6 +207,7 @@ PG_KEYWORD("in", IN_P, RESERVED_KEYWORD, BARE_LABEL)
PG_KEYWORD("include", INCLUDE, UNRESERVED_KEYWORD, BARE_LABEL)
PG_KEYWORD("including", INCLUDING, UNRESERVED_KEYWORD, BARE_LABEL)
PG_KEYWORD("increment", INCREMENT, UNRESERVED_KEYWORD, BARE_LABEL)
+PG_KEYWORD("incremental", INCREMENTAL, UNRESERVED_KEYWORD, BARE_LABEL)
PG_KEYWORD("indent", INDENT, UNRESERVED_KEYWORD, BARE_LABEL)
PG_KEYWORD("index", INDEX, UNRESERVED_KEYWORD, BARE_LABEL)
PG_KEYWORD("indexes", INDEXES, UNRESERVED_KEYWORD, BARE_LABEL)
--
2.25.1
--Multipart=_Mon__4_Mar_2024_11_58_46_+0900_UaponF/qQhQrVCFt
Content-Type: text/x-diff;
name="v30-0002-Add-relisivm-column-to-pg_class-system-catalog.patch"
Content-Disposition: attachment;
filename="v30-0002-Add-relisivm-column-to-pg_class-system-catalog.patch"
Content-Transfer-Encoding: 7bit
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v1] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 ++--
src/backend/commands/prepare.c | 43 +++++++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++++++---------
src/test/regress/sql/prepare.sql | 2 +-
4 files changed, 72 insertions(+), 25 deletions(-)
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 6af5ac1da18..535c4770095 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 34b6410d6a2..11b330ab73d 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--biRhQYMBrGQlqcCN--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v3] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 1 +
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 26 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index c27e9529bb6..ac967687796 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -36,6 +36,7 @@ REGRESS = \
parallel \
plancache \
squashing \
+ prepare \
cleanup \
oldextversions
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 9d78cb88b7d..c9e1d446d39 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -58,6 +58,7 @@ tests += {
'parallel',
'plancache',
'squashing',
+ 'prepare',
'cleanup',
'oldextversions',
],
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--xBRJzsblsXcilRYX--
^ permalink raw reply [nested|flat] 496+ messages in thread
* [PATCH v2] Cleanup explicit PREPARE query strings
@ 2025-12-24 14:31 Julien Rouhaud <[email protected]>
0 siblings, 0 replies; 496+ messages in thread
From: Julien Rouhaud @ 2025-12-24 14:31 UTC (permalink / raw)
When a multi statements query string contains one PREPARE statement (or
multiple), the whole query string was saved in the cached plan. This is
wasteful as that string can be artitrarily big, but it can also confusing as
some other parts like pg_prepared_statements will output the saved query string
as-is.
This commit changes this behavior and only stores the part of the query string
that correspond to any given PREPARED statement, similarly to how it's already
done in pg_stat_statements.
---
contrib/auto_explain/t/001_auto_explain.pl | 8 +--
contrib/pg_stat_statements/Makefile | 2 +-
.../pg_stat_statements/expected/prepare.out | 53 +++++++++++++++++++
contrib/pg_stat_statements/meson.build | 1 +
contrib/pg_stat_statements/sql/prepare.sql | 15 ++++++
src/backend/commands/prepare.c | 45 ++++++++++++++--
src/test/regress/expected/prepare.out | 44 +++++++++------
src/test/regress/sql/prepare.sql | 2 +-
8 files changed, 143 insertions(+), 27 deletions(-)
create mode 100644 contrib/pg_stat_statements/expected/prepare.out
create mode 100644 contrib/pg_stat_statements/sql/prepare.sql
diff --git a/contrib/auto_explain/t/001_auto_explain.pl b/contrib/auto_explain/t/001_auto_explain.pl
index 5f673bd14c1..f2d8625f8bb 100644
--- a/contrib/auto_explain/t/001_auto_explain.pl
+++ b/contrib/auto_explain/t/001_auto_explain.pl
@@ -60,7 +60,7 @@ $log_contents = query_log($node,
like(
$log_contents,
- qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1;/,
+ qr/Query Text: PREPARE get_proc\(name\) AS SELECT \* FROM pg_proc WHERE proname = \$1/,
"prepared query text logged, text mode");
like(
@@ -82,7 +82,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
like(
@@ -98,7 +98,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1;/,
+ qr/Query Text: PREPARE get_type\(name\) AS SELECT \* FROM pg_type WHERE typname = \$1/,
"prepared query text logged, text mode");
unlike(
@@ -164,7 +164,7 @@ $log_contents = query_log(
like(
$log_contents,
- qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1;"/,
+ qr/"Query Text": "PREPARE get_class\(name\) AS SELECT \* FROM pg_class WHERE relname = \$1"/,
"prepared query text logged, json mode");
like(
diff --git a/contrib/pg_stat_statements/Makefile b/contrib/pg_stat_statements/Makefile
index fe0478ac552..9a6bda90070 100644
--- a/contrib/pg_stat_statements/Makefile
+++ b/contrib/pg_stat_statements/Makefile
@@ -21,7 +21,7 @@ LDFLAGS_SL += $(filter -lm, $(LIBS))
REGRESS_OPTS = --temp-config $(top_srcdir)/contrib/pg_stat_statements/pg_stat_statements.conf
REGRESS = select dml cursors utility level_tracking planning \
user_activity wal entry_timestamp privileges extended \
- parallel plancache cleanup oldextversions squashing
+ parallel plancache cleanup oldextversions squashing prepare
# Disabled because these tests require "shared_preload_libraries=pg_stat_statements",
# which typical installcheck users do not have (e.g. buildfarm clients).
NO_INSTALLCHECK = 1
diff --git a/contrib/pg_stat_statements/expected/prepare.out b/contrib/pg_stat_statements/expected/prepare.out
new file mode 100644
index 00000000000..010e289c1b0
--- /dev/null
+++ b/contrib/pg_stat_statements/expected/prepare.out
@@ -0,0 +1,53 @@
+-- Tests for PREPARE
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+ ?column?
+----------
+ 1
+(1 row)
+
+ ?column? | ?column?
+----------+----------
+ 1 | 1
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | SELECT $1
+ 1 | 1 | SELECT $1, $2
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+ t
+---
+ t
+(1 row)
+
+EXECUTE p1;
+ ?column?
+----------
+ 1
+(1 row)
+
+EXECUTE p2(0);
+ ?column?
+----------
+ 0
+(1 row)
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+ calls | rows | query
+-------+------+----------------------------------------------------
+ 1 | 1 | PREPARE p1 AS SELECT 1
+ 1 | 1 | PREPARE p2(int) AS SELECT 2 * $1
+ 1 | 1 | SELECT pg_stat_statements_reset() IS NOT NULL AS t
+(3 rows)
+
diff --git a/contrib/pg_stat_statements/meson.build b/contrib/pg_stat_statements/meson.build
index 079a8eb5afc..65258651bd2 100644
--- a/contrib/pg_stat_statements/meson.build
+++ b/contrib/pg_stat_statements/meson.build
@@ -59,6 +59,7 @@ tests += {
'cleanup',
'oldextversions',
'squashing',
+ 'prepare',
],
'regress_args': ['--temp-config', files('pg_stat_statements.conf')],
# Disabled because these tests require
diff --git a/contrib/pg_stat_statements/sql/prepare.sql b/contrib/pg_stat_statements/sql/prepare.sql
new file mode 100644
index 00000000000..a6bc5de4430
--- /dev/null
+++ b/contrib/pg_stat_statements/sql/prepare.sql
@@ -0,0 +1,15 @@
+-- Tests for PREPARE
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+-- Test that prepared statements in a multi-query string behaves as expected
+SELECT 1\;PREPARE p1 AS SELECT 1\; PREPARE p2(int) AS SELECT 2 * $1\; SELECT 1, 1;
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
+
+SELECT pg_stat_statements_reset() IS NOT NULL AS t;
+
+EXECUTE p1;
+EXECUTE p2(0);
+
+SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
diff --git a/src/backend/commands/prepare.c b/src/backend/commands/prepare.c
index 5b86a727587..b8fe3dd5302 100644
--- a/src/backend/commands/prepare.c
+++ b/src/backend/commands/prepare.c
@@ -27,6 +27,7 @@
#include "commands/prepare.h"
#include "funcapi.h"
#include "nodes/nodeFuncs.h"
+#include "nodes/queryjumble.h"
#include "parser/parse_coerce.h"
#include "parser/parse_collate.h"
#include "parser/parse_expr.h"
@@ -64,6 +65,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
Oid *argtypes = NULL;
int nargs;
List *query_list;
+ const char *new_query;
/*
* Disallow empty-string statement name (conflicts with protocol-level
@@ -80,14 +82,49 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
*/
rawstmt = makeNode(RawStmt);
rawstmt->stmt = stmt->query;
- rawstmt->stmt_location = stmt_location;
- rawstmt->stmt_len = stmt_len;
+
+ /*
+ * Extract the query text if possible.
+ *
+ * If we have a statement location, we can extract the relevant part of the
+ * possibly multi-statement query string. If not just use what we were
+ * given.
+ */
+ if (stmt_location < 0)
+ {
+ rawstmt->stmt_location = stmt_location;
+ rawstmt->stmt_len = stmt_len;
+ new_query = pstate->p_sourcetext;
+ }
+ else
+ {
+ const char *cleaned;
+ char *tmp;
+
+ rawstmt->stmt_len = stmt_len;
+ cleaned = CleanQuerytext(pstate->p_sourcetext, &stmt_location,
+ &rawstmt->stmt_len);
+
+ if (rawstmt->stmt_len == 0)
+ rawstmt->stmt_len = strlen(cleaned);
+
+ /*
+ * CleanQuerytext() removes any leading whitespace and returns a
+ * pointer to the first actual character, so the cleaned query string
+ * is guaranteed to start at offset 0.
+ */
+ rawstmt->stmt_location = 0;
+ tmp = palloc(rawstmt->stmt_len + 1);
+ strlcpy(tmp, cleaned, rawstmt->stmt_len + 1);
+
+ new_query = tmp;
+ }
/*
* Create the CachedPlanSource before we do parse analysis, since it needs
* to see the unmodified raw parse tree.
*/
- plansource = CreateCachedPlan(rawstmt, pstate->p_sourcetext,
+ plansource = CreateCachedPlan(rawstmt, new_query,
CreateCommandTag(stmt->query));
/* Transform list of TypeNames to array of type OIDs */
@@ -116,7 +153,7 @@ PrepareQuery(ParseState *pstate, PrepareStmt *stmt,
* information about unknown parameters to be deduced from context.
* Rewrite the query. The result could be 0, 1, or many queries.
*/
- query_list = pg_analyze_and_rewrite_varparams(rawstmt, pstate->p_sourcetext,
+ query_list = pg_analyze_and_rewrite_varparams(rawstmt, new_query,
&argtypes, &nargs, NULL);
/* Finish filling in the CachedPlanSource */
diff --git a/src/test/regress/expected/prepare.out b/src/test/regress/expected/prepare.out
index 5815e17b39c..c645a4e5d0e 100644
--- a/src/test/regress/expected/prepare.out
+++ b/src/test/regress/expected/prepare.out
@@ -6,7 +6,17 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+-----------+-----------------+--------------
(0 rows)
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
+ ?column?
+----------
+ bingo
+(1 row)
+
+ ?column?
+----------
+ 42
+(1 row)
+
EXECUTE q1;
a
---
@@ -14,9 +24,9 @@ EXECUTE q1;
(1 row)
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 1 AS a; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 1 AS a | {} | {integer}
(1 row)
-- should fail
@@ -33,18 +43,18 @@ EXECUTE q1;
PREPARE q2 AS SELECT 2 AS b;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q1 | PREPARE q1 AS SELECT 2; | {} | {integer}
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q1 | PREPARE q1 AS SELECT 2 | {} | {integer}
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(2 rows)
-- sql92 syntax
DEALLOCATE PREPARE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
- name | statement | parameter_types | result_types
-------+------------------------------+-----------------+--------------
- q2 | PREPARE q2 AS SELECT 2 AS b; | {} | {integer}
+ name | statement | parameter_types | result_types
+------+-----------------------------+-----------------+--------------
+ q2 | PREPARE q2 AS SELECT 2 AS b | {} | {integer}
(1 row)
DEALLOCATE PREPARE q2;
@@ -168,20 +178,20 @@ SELECT name, statement, parameter_types, result_types FROM pg_prepared_statement
------+------------------------------------------------------------------+----------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------
q2 | PREPARE q2(text) AS +| {text} | {name,boolean,boolean}
| SELECT datname, datistemplate, datallowconn +| |
- | FROM pg_database WHERE datname = $1; | |
+ | FROM pg_database WHERE datname = $1 | |
q3 | PREPARE q3(text, int, float, boolean, smallint) AS +| {text,integer,"double precision",boolean,smallint} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE string4 = $1 AND (four = $2 OR+| |
| ten = $3::bigint OR true = $4 OR odd = $5::int) +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q5 | PREPARE q5(int, text) AS +| {integer,text} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
| SELECT * FROM tenk1 WHERE unique1 = $1 OR stringu1 = $2 +| |
- | ORDER BY unique1; | |
+ | ORDER BY unique1 | |
q6 | PREPARE q6 AS +| {integer,name} | {integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,integer,name,name,name}
- | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2; | |
+ | SELECT * FROM tenk1 WHERE unique1 = $1 AND stringu1 = $2 | |
q7 | PREPARE q7(unknown) AS +| {path} | {text,path}
- | SELECT * FROM road WHERE thepath = $1; | |
+ | SELECT * FROM road WHERE thepath = $1 | |
q8 | PREPARE q8 AS +| {integer,name} |
- | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1; | |
+ | UPDATE tenk1 SET stringu1 = $2 WHERE unique1 = $1 | |
(6 rows)
-- test DEALLOCATE ALL;
diff --git a/src/test/regress/sql/prepare.sql b/src/test/regress/sql/prepare.sql
index c6098dc95ce..0e7fe44725e 100644
--- a/src/test/regress/sql/prepare.sql
+++ b/src/test/regress/sql/prepare.sql
@@ -4,7 +4,7 @@
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
-PREPARE q1 AS SELECT 1 AS a;
+SELECT 'bingo'\; PREPARE q1 AS SELECT 1 AS a \; SELECT 42;
EXECUTE q1;
SELECT name, statement, parameter_types, result_types FROM pg_prepared_statements;
--
2.52.0
--soawQPnav6xcFtRL--
^ permalink raw reply [nested|flat] 496+ messages in thread
end of thread, other threads:[~2025-12-24 14:31 UTC | newest]
Thread overview: 496+ messages (download: mbox mbox.gz follow: Atom feed)
-- links below jump to the message on this page --
2019-12-20 01:05 [PATCH v30 01/11] Add a syntax to create Incrementally Maintainable Materialized Views Yugo Nagata <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v1] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v3] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
2025-12-24 14:31 [PATCH v2] Cleanup explicit PREPARE query strings Julien Rouhaud <[email protected]>
This inbox is served by agora; see mirroring instructions
for how to clone and mirror all data and code used for this inbox