use a utility method to get event size

pull/3491/head
SaitTalhaNisanci 2020-02-12 11:46:38 +03:00
parent 71f1aa48a3
commit a7e735a648
1 changed files with 15 additions and 4 deletions

View File

@ -618,6 +618,7 @@ static bool HasDependentJobs(Job *mainJob);
static void ExtractParametersForRemoteExecution(ParamListInfo paramListInfo,
Oid **parameterTypes,
const char ***parameterValues);
static int GetEventSetSize(List *sessionList);
/*
* AdaptiveExecutor is called via CitusExecScan on the
@ -2007,8 +2008,7 @@ RunDistributedExecution(DistributedExecution *execution)
{
bool cancellationReceived = false;
/* additional 2 is for postmaster and latch */
int eventSetSize = list_length(execution->sessionList) + 2;
int eventSetSize = GetEventSetSize(execution->sessionList);
/* always (re)build the wait event set the first time */
execution->connectionSetChanged = true;
@ -2045,7 +2045,7 @@ RunDistributedExecution(DistributedExecution *execution)
execution->waitEventSet = BuildWaitEventSet(execution->sessionList);
/* recalculate (and allocate) since the sessions have changed */
eventSetSize = list_length(execution->sessionList) + 2;
eventSetSize = GetEventSetSize(execution->sessionList);
events = palloc0(eventSetSize * sizeof(WaitEvent));
@ -3766,7 +3766,7 @@ static WaitEventSet *
BuildWaitEventSet(List *sessionList)
{
/* additional 2 is for postmaster and latch */
int eventSetSize = list_length(sessionList) + 2;
int eventSetSize = GetEventSetSize(sessionList);
WaitEventSet *waitEventSet =
CreateWaitEventSet(CurrentMemoryContext, eventSetSize);
@ -3807,6 +3807,17 @@ BuildWaitEventSet(List *sessionList)
}
/*
* GetEventSetSize returns the event set size.
*/
static int
GetEventSetSize(List *sessionList)
{
/* additional 2 is for postmaster and latch */
return list_length(sessionList) + 2;
}
/*
* UpdateWaitEventSetFlags modifies the given waitEventSet with the wait flags
* for connections in the sessionList.