Judge whether commands need to be dispatched to QEs in QD (#314)

apache · Nov 24, 2023 · 57a5cf8 · 57a5cf8
1 parent e0ed138
commit 57a5cf8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/backend/cdb/dispatcher/cdbdisp_query.c b/src/backend/cdb/dispatcher/cdbdisp_query.c
@@ -312,6 +312,10 @@ CdbDispatchSetCommand(const char *strCommand, bool cancelOnError)
 	int		queryTextLength;
 	ListCell   *le;
 	ErrorData *qeError = NULL;
+	int flags = DF_NONE;
+
+	if (CdbNeedDispatchCommand_hook && !CdbNeedDispatchCommand_hook(strCommand, &flags, NULL, NULL))
+		return;
 
 	elog((Debug_print_full_dtm ? LOG : DEBUG5),
 		 "CdbDispatchSetCommand for command = '%s'",