Explicitly forbid non-top-level WAIT FOR execution

Previously we were relying on a snapshot-based check to detect invalid
execution contexts.  However, when WAIT FOR is wrapped into a stored
procedure or a DO block, it could pass this check, causing an error
elsewhere.

This commit implements an explicit isTopLevel check to reject WAIT FOR
when called from within a function, procedure, or DO block.  The
isTopLevel check catches these cases early with a clear error message,
matching the pattern used by other utility commands like VACUUM and
REINDEX.  The snapshot check is retained for the remaining case:
top-level execution within a transaction block using an isolation level
higher than READ COMMITTED.

Also adds tests for WAIT FOR LSN wrapped in a procedure and DO block,
complementing the existing test that uses a function wrapper.  Relevant
documentation paragraph is also added.

Reported-by: Satyanarayana Narlapuram <[email protected]>
Discussion: 
https://postgr.es/m/CAHg%2BQDcN-n3NUqgRtj%3DBQb9fFQmH8-DeEROCr%3DPDbw_BBRKOYA%40mail.gmail.com
Author: Satyanarayana Narlapuram <[email protected]>
Reviewed-by: Alexander Korotkov <[email protected]>
Reviewed-by: Xuneng Zhou <[email protected]>

Branch
------
master

Details
-------
https://git.postgresql.org/pg/commitdiff/a8b61c23c5ff980312026a05cab5502aa31fd710

Modified Files
--------------
doc/src/sgml/ref/wait_for.sgml          |  8 ++++++++
src/backend/commands/wait.c             | 23 +++++++++++++++++------
src/backend/tcop/utility.c              |  3 ++-
src/include/commands/wait.h             |  3 ++-
src/test/recovery/t/049_wait_for_lsn.pl | 30 +++++++++++++++++++++++++++---
5 files changed, 56 insertions(+), 11 deletions(-)

Reply via email to