NotificationsYou must be signed in to change notification settings
Fork38
Star158

Commit174a786

Oleg Tselebrovskiy

committed

Add ExecutorStart hook to track top-level queryId for prepared queries

and more precisely in general

1 parent92097fd commit174a786Copy full SHA for 174a786

File tree

1 file changed

+36

-0

lines changed

pg_wait_sampling.c

1 file changed

+36

-0

lines changed

`‎pg_wait_sampling.c`

Lines changed: 36 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,7 @@ void_PG_init(void);`
`44`	`44`	`staticboolshmem_initialized= false;`
`45`	`45`
`46`	`46`	`/* Hooks */`
	`47`	`+staticExecutorStart_hook_typeprev_ExecutorStart=NULL;`
`47`	`48`	`staticExecutorEnd_hook_typeprev_ExecutorEnd=NULL;`
`48`	`49`	`staticplanner_hook_typeplanner_hook_next=NULL;`
`49`	`50`
`@@ -67,6 +68,7 @@ static PlannedStmt pgws_planner_hook(Query parse,`
`67`	`68`	`constchar*query_string,`
`68`	`69`	`#endif`
`69`	`70`	`intcursorOptions,ParamListInfoboundParams);`
	`71`	`+staticvoidpgws_ExecutorStart(QueryDesc*queryDesc,inteflags);`
`70`	`72`	`staticvoidpgws_ExecutorEnd(QueryDesc*queryDesc);`
`71`	`73`
`72`	`74`	`/*`
`@@ -402,6 +404,8 @@ _PG_init(void)`
`402`	`404`	`shmem_startup_hook=pgws_shmem_startup;`
`403`	`405`	`planner_hook_next=planner_hook;`
`404`	`406`	`planner_hook=pgws_planner_hook;`
	`407`	`+prev_ExecutorStart=ExecutorStart_hook;`
	`408`	`+ExecutorStart_hook=pgws_ExecutorStart;`
`405`	`409`	`prev_ExecutorEnd=ExecutorEnd_hook;`
`406`	`410`	`ExecutorEnd_hook=pgws_ExecutorEnd;`
`407`	`411`	`}`
`@@ -906,6 +910,38 @@ pgws_planner_hook(Query *parse,`
`906`	`910`	`cursorOptions,boundParams);`
`907`	`911`	`}`
`908`	`912`
	`913`	`+/*`
	`914`	`+ * ExecutorStart hook: save queryId for collector`
	`915`	`+ */`
	`916`	`+staticvoid`
	`917`	`+pgws_ExecutorStart(QueryDesc*queryDesc,inteflags)`
	`918`	`+{`
	`919`	`+inti;`
	`920`	`+`
	`921`	`+if (MyProc)`
	`922`	`+{`
	`923`	`+i=MyProc-ProcGlobal->allProcs;`
	`924`	`+#ifPG_VERSION_NUM >=110000`
	`925`	`+/*`
	`926`	`+ * since we depend on queryId we need to check that its size`
	`927`	`+ * is uint64 as we coded in pg_wait_sampling`
	`928`	`+ */`
	`929`	`+StaticAssertExpr(sizeof(queryDesc->plannedstmt->queryId)==sizeof(uint64),`
	`930`	`+"queryId size is not uint64");`
	`931`	`+#else`
	`932`	`+StaticAssertExpr(sizeof(queryDesc->plannedstmt->queryId)==sizeof(uint32),`
	`933`	`+"queryId size is not uint32");`
	`934`	`+#endif`
	`935`	`+if (!pgws_proc_queryids[i])`
	`936`	`+pgws_proc_queryids[i]=queryDesc->plannedstmt->queryId;`
	`937`	`+}`
	`938`	`+`
	`939`	`+if (prev_ExecutorStart)`
	`940`	`+prev_ExecutorStart(queryDesc,eflags);`
	`941`	`+else`
	`942`	`+standard_ExecutorStart(queryDesc,eflags);`
	`943`	`+}`
	`944`	`+`
`909`	`945`	`/*`
`910`	`946`	`* ExecutorEnd hook: clear queryId`
`911`	`947`	`*/`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit174a786

File tree

1 file changed

1 file changed

`‎pg_wait_sampling.c`

0 commit comments