diff --git a/src/test/modules/test_session_hooks/test_session_hooks.c b/src/test/modules/test_session_hooks/test_session_hooks.c index d047c5d219..a90c373c69 100644 --- a/src/test/modules/test_session_hooks/test_session_hooks.c +++ b/src/test/modules/test_session_hooks/test_session_hooks.c @@ -13,6 +13,7 @@ */ #include "postgres.h" +#include "access/parallel.h" #include "access/xact.h" #include "commands/dbcommands.h" #include "executor/spi.h" @@ -89,6 +90,10 @@ sample_session_start_hook(void) if (!OidIsValid(MyDatabaseId)) return; + /* no parallel workers */ + if (IsParallelWorker()) + return; + register_session_hook("START"); } @@ -107,6 +112,10 @@ sample_session_end_hook(void) if (!OidIsValid(MyDatabaseId)) return; + /* no parallel workers */ + if (IsParallelWorker()) + return; + register_session_hook("END"); }