statement_timeout.diff

text/x-patch

Filename: statement_timeout.diff

Type: text/x-patch

Part: 0

Message: Re: BUG #16035: STATEMENT_TIMEOUT not working when we have single quote usage inside CTE which is used in inner sql

Patch

Format: unified

File	+	−
src/backend/tcop/postgres.c	2	1

diff --git a/src/backend/tcop/postgres.c b/src/backend/tcop/postgres.c
index e8d8e6f828..0ba6fa9d5e 100644
--- a/src/backend/tcop/postgres.c
+++ b/src/backend/tcop/postgres.c
@@ -4701,7 +4701,8 @@ enable_statement_timeout(void)
 
 	if (StatementTimeout > 0)
 	{
-		if (!stmt_timeout_active)
+		if (!doing_extended_query_message ||
+			(!stmt_timeout_active && doing_extended_query_message))
 		{
 			enable_timeout_after(STATEMENT_TIMEOUT, StatementTimeout);
 			stmt_timeout_active = true;