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

Defer initial PQflush to main loop in WaitForAllConnections

pull/1737/head
Marco Slot 2017-10-28 15:54:18 +02:00
parent d6dadb1b25
commit e6e6897499
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/backend/distributed/connection/remote_commands.c
View File

@ -848,7 +848,14 @@ WaitForAllConnections(List *connectionList, bool raiseInterrupts)
}
}
if (event->events & WL_SOCKET_READABLE)
/*
* Check whether the connection is done is the socket is either readable
* or writable. If it was only writable, we performed a PQflush which
* might have read from the socket, meaning we may not see the socket
* becoming readable again, so better to check it now.
*/
if (event->events & WL_SOCKET_READABLE ||
event->events & WL_SOCKET_WRITEABLE)
{
int receiveStatus = PQconsumeInput(connection->pgConn);
if (receiveStatus == 0)
@ -907,8 +914,8 @@ WaitForAllConnections(List *connectionList, bool raiseInterrupts)
/*
* BuildWaitEventSet creates a WaitEventSet for the given array of connections
* which can be used to wait for any of the sockets to become read-ready, or
* write-ready in case there is data to send.
* which can be used to wait for any of the sockets to become read-ready or
* write-ready.
*/
static WaitEventSet *
BuildWaitEventSet(MultiConnection **allConnections, int totalConnectionCount,
@ -926,14 +933,7 @@ BuildWaitEventSet(MultiConnection **allConnections, int totalConnectionCount,
{
MultiConnection *connection = allConnections[connectionIndex];
int socket = PQsocket(connection->pgConn);
int eventMask = WL_SOCKET_READABLE;
int sendStatus = PQflush(connection->pgConn);
if (sendStatus == 1)
{
/* we have data to send, wake up when the socket is ready to write */
eventMask = WL_SOCKET_READABLE | WL_SOCKET_WRITEABLE;
}
int eventMask = WL_SOCKET_READABLE | WL_SOCKET_WRITEABLE;
AddWaitEventToSet(waitEventSet, eventMask, socket, NULL, (void *) connection);
}