diff options
| -rw-r--r-- | src/backend/executor/execMain.c | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/backend/executor/execMain.c b/src/backend/executor/execMain.c index 7042ca6c600..713cf3e802c 100644 --- a/src/backend/executor/execMain.c +++ b/src/backend/executor/execMain.c @@ -50,6 +50,7 @@ #include "foreign/fdwapi.h" #include "mb/pg_wchar.h" #include "miscadmin.h" +#include "nodes/queryjumble.h" #include "parser/parse_relation.h" #include "rewrite/rewriteHandler.h" #include "tcop/utility.h" @@ -295,6 +296,9 @@ ExecutorRun(QueryDesc *queryDesc, ScanDirection direction, uint64 count, bool execute_once) { + /* If enabled, the query ID should be set. */ + Assert(!IsQueryIdEnabled() || pgstat_get_my_query_id() != 0); + if (ExecutorRun_hook) (*ExecutorRun_hook) (queryDesc, direction, count, execute_once); else @@ -403,6 +407,9 @@ standard_ExecutorRun(QueryDesc *queryDesc, void ExecutorFinish(QueryDesc *queryDesc) { + /* If enabled, the query ID should be set. */ + Assert(!IsQueryIdEnabled() || pgstat_get_my_query_id() != 0); + if (ExecutorFinish_hook) (*ExecutorFinish_hook) (queryDesc); else @@ -463,6 +470,9 @@ standard_ExecutorFinish(QueryDesc *queryDesc) void ExecutorEnd(QueryDesc *queryDesc) { + /* If enabled, the query ID should be set. */ + Assert(!IsQueryIdEnabled() || pgstat_get_my_query_id() != 0); + if (ExecutorEnd_hook) (*ExecutorEnd_hook) (queryDesc); else |