hcq/openGauss-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add plannerHook

Browse Source
This commit is contained in:
ganyang
2023-01-03 11:54:40 +08:00
parent ea4f17cc42
commit 255c419497
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/gausskernel/process/tcop/postgres.cpp
View File

@ -1289,8 +1289,12 @@ PlannedStmt* pg_plan_query(Query* querytree, int cursorOptions, ParamListInfo bo
if (!OidIsValid(lc_replan_nodegroup))
check_query_acl(querytree);
/* call the optimizer */
plan = planner(querytree, cursorOptions, boundParams);
if (u_sess->hook_cxt.plannerHook != NULL) {
plan = ((plannerFunc)(u_sess->hook_cxt.plannerHook))(querytree, cursorOptions, boundParams);
} else {
/* call the optimizer */
plan = planner(querytree, cursorOptions, boundParams);
}
PGSTAT_END_TIME_RECORD(PLAN_TIME);

1
src/include/knl/knl_session.h
View File

@ -2711,6 +2711,7 @@ typedef struct knl_u_hook_context {
void *standardProcessUtilityHook;
void *aggIsSupportedHook;
void *searchFuncHook;
void *plannerHook;
} knl_u_hook_context;
/* PBE message flag */
typedef enum {

2
src/include/optimizer/planner.h
View File

@ -209,4 +209,6 @@ extern bool check_stream_for_loop_fetch(Portal portal);
extern bool IsPlanForPartitionScan(Plan* plan);
extern bool queryIsReadOnly(Query* query);
typedef PlannedStmt* (*plannerFunc)(Query* parse, int cursorOptions, ParamListInfo boundParams);
#endif /* PLANNER_H */