external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

w

remove_socket_check
Onder Kalaci 2022-11-18 16:58:23 +01:00
parent 11ff4a1faf
commit 1e5096e29d
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/backend/distributed/executor/adaptive_executor.c
View File

@ -3326,10 +3326,10 @@ OpenNewConnections(WorkerPool *workerPool, int newConnectionCount,
newConnectionsList = lappend(newConnectionsList, session);
}
#if PG_VERSION_NUM >= PG_VERSION_15
bool socketClosedOnly = true;
RebuildWaitEventSet(workerPool->distributedExecution, socketClosedOnly);
/*
* Before using this connection in the distributed execution, we check
* whether the remote connection is closed/lost. This is common
@ -3344,7 +3344,6 @@ OpenNewConnections(WorkerPool *workerPool, int newConnectionCount,
workerPool->distributedExecution->eventSetSize,
WAIT_EVENT_CLIENT_READ);
#if PG_VERSION_NUM >= PG_VERSION_15
ProcessWaitEventsForSocketClosed(workerPool->distributedExecution->events,
eventCount);
#endif
@ -5490,8 +5489,10 @@ BuildWaitEventSet(List *sessionList, bool socketClosedOnly)
NULL);
if (!socketClosedOnly)
{
CitusAddWaitEventSetToSet(waitEventSet, WL_LATCH_SET, PGINVALID_SOCKET, MyLatch,
NULL);
}
return waitEventSet;
}
@ -5503,8 +5504,8 @@ BuildWaitEventSet(List *sessionList, bool socketClosedOnly)
* to the waitEventSet.
*/
static void
AddSessionToWaitEventSet(WorkerSession *session, WaitEventSet *waitEventSet, bool
socketClosedOnly)
AddSessionToWaitEventSet(WorkerSession *session, WaitEventSet *waitEventSet,
bool socketClosedOnly)
{
MultiConnection *connection = session->connection;
@ -5531,9 +5532,15 @@ AddSessionToWaitEventSet(WorkerSession *session, WaitEventSet *waitEventSet, boo
if (socketClosedOnly)
{
#if PG_VERSION_NUM >= PG_VERSION_15
waitEventSetIndex = CitusAddWaitEventSetToSet(waitEventSet, WL_SOCKET_CLOSED,
sock,
NULL, (void *) session);
#else
/* socketClosedOnly is only true for */
Assert (false);
#endif
}
else
{