diff options
Diffstat (limited to 'src/backend/utils/activity/wait_event_funcs.c')
| -rw-r--r-- | src/backend/utils/activity/wait_event_funcs.c | 31 |
1 files changed, 28 insertions, 3 deletions
diff --git a/src/backend/utils/activity/wait_event_funcs.c b/src/backend/utils/activity/wait_event_funcs.c index ba244c2cf36..fa8bc05c0c7 100644 --- a/src/backend/utils/activity/wait_event_funcs.c +++ b/src/backend/utils/activity/wait_event_funcs.c @@ -48,7 +48,7 @@ pg_get_wait_events(PG_FUNCTION_ARGS) #define PG_GET_WAIT_EVENTS_COLS 3 ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo; char **waiteventnames; - int nbextwaitevents; + int nbwaitevents; /* Build tuplestore to hold the result rows */ InitMaterializedSRF(fcinfo, 0); @@ -67,9 +67,10 @@ pg_get_wait_events(PG_FUNCTION_ARGS) } /* Handle custom wait events for extensions */ - waiteventnames = GetWaitEventExtensionNames(&nbextwaitevents); + waiteventnames = GetWaitEventCustomNames(PG_WAIT_EXTENSION, + &nbwaitevents); - for (int idx = 0; idx < nbextwaitevents; idx++) + for (int idx = 0; idx < nbwaitevents; idx++) { StringInfoData buf; Datum values[PG_GET_WAIT_EVENTS_COLS] = {0}; @@ -89,5 +90,29 @@ pg_get_wait_events(PG_FUNCTION_ARGS) tuplestore_putvalues(rsinfo->setResult, rsinfo->setDesc, values, nulls); } + /* Likewise for injection points */ + waiteventnames = GetWaitEventCustomNames(PG_WAIT_INJECTIONPOINT, + &nbwaitevents); + + for (int idx = 0; idx < nbwaitevents; idx++) + { + StringInfoData buf; + Datum values[PG_GET_WAIT_EVENTS_COLS] = {0}; + bool nulls[PG_GET_WAIT_EVENTS_COLS] = {0}; + + + values[0] = CStringGetTextDatum("InjectionPoint"); + values[1] = CStringGetTextDatum(waiteventnames[idx]); + + initStringInfo(&buf); + appendStringInfo(&buf, + "Waiting for injection point \"%s\"", + waiteventnames[idx]); + + values[2] = CStringGetTextDatum(buf.data); + + tuplestore_putvalues(rsinfo->setResult, rsinfo->setDesc, values, nulls); + } + return (Datum) 0; } |