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

Keep more statistics about connection establishment times 

When DEBUG4 enabled, Citus now prints per connection establishment
time.
pull/4892/head
Onder Kalaci 2021-04-01 15:33:02 +02:00
parent 0a060b327b
commit 5482d5822f
3 changed files with 34 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/backend/distributed/connection/connection_management.c
View File

@ -940,8 +940,7 @@ FinishConnectionListEstablishment(List *multiConnectionList)
*/
if (connectionState->phase == MULTI_CONNECTION_PHASE_CONNECTED)
{
connectionState->connection->connectionState =
MULTI_CONNECTION_CONNECTED;
MarkConnectionConnected(connectionState->connection);
}
}
}
@ -1142,7 +1141,7 @@ StartConnectionEstablishment(MultiConnection *connection, ConnectionHashKey *key
connection->pgConn = PQconnectStartParams((const char **) entry->keywords,
(const char **) entry->values,
false);
connection->connectionStart = GetCurrentTimestamp();
INSTR_TIME_SET_CURRENT(connection->connectionEstablishmentStart);
connection->connectionId = connectionId++;
/*
@ -1344,8 +1343,8 @@ ShouldShutdownConnection(MultiConnection *connection, const int cachedConnection
PQstatus(connection->pgConn) != CONNECTION_OK ||
!RemoteTransactionIdle(connection) ||
(MaxCachedConnectionLifetime >= 0 &&
TimestampDifferenceExceeds(connection->connectionStart, GetCurrentTimestamp(),
MaxCachedConnectionLifetime));
MillisecondsToTimeout(connection->connectionEstablishmentStart,
MaxCachedConnectionLifetime) <= 0);
}
@ -1400,3 +1399,20 @@ RemoteTransactionIdle(MultiConnection *connection)
return PQtransactionStatus(connection->pgConn) == PQTRANS_IDLE;
}
/*
* MarkConnectionConnected is a helper function which sets the connection
* connectionState to MULTI_CONNECTION_CONNECTED, and also updates connection
* establishment time when necessary.
*/
void
MarkConnectionConnected(MultiConnection *connection)
{
connection->connectionState = MULTI_CONNECTION_CONNECTED;
if (INSTR_TIME_IS_ZERO(connection->connectionEstablishmentEnd))
{
INSTR_TIME_SET_CURRENT(connection->connectionEstablishmentEnd);
}
}

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

@ -2945,8 +2945,6 @@ ConnectionStateMachine(WorkerSession *session)
HandleMultiConnectionSuccess(session);
UpdateConnectionWaitFlags(session,
WL_SOCKET_READABLE | WL_SOCKET_WRITEABLE);
connection->connectionState = MULTI_CONNECTION_CONNECTED;
break;
}
else if (status == CONNECTION_BAD)
@ -2988,8 +2986,6 @@ ConnectionStateMachine(WorkerSession *session)
UpdateConnectionWaitFlags(session,
WL_SOCKET_READABLE | WL_SOCKET_WRITEABLE);
connection->connectionState = MULTI_CONNECTION_CONNECTED;
/* we should have a valid socket */
Assert(PQsocket(connection->pgConn) != -1);
}
@ -3116,7 +3112,8 @@ ConnectionStateMachine(WorkerSession *session)
/*
* HandleMultiConnectionSuccess logs the established connection and updates connection's state.
* HandleMultiConnectionSuccess logs the established connection and updates
* connection's state.
*/
static void
HandleMultiConnectionSuccess(WorkerSession *session)
@ -3124,10 +3121,15 @@ HandleMultiConnectionSuccess(WorkerSession *session)
MultiConnection *connection = session->connection;
WorkerPool *workerPool = session->workerPool;
MarkConnectionConnected(connection);
ereport(DEBUG4, (errmsg("established connection to %s:%d for "
"session %ld",
"session %ld in %ld msecs",
connection->hostname, connection->port,
session->sessionId)));
session->sessionId,
MillisecondsBetweenTimestamps(
connection->connectionEstablishmentStart,
connection->connectionEstablishmentEnd))));
workerPool->activeConnectionCount++;
workerPool->idleConnectionCount++;

6
src/include/distributed/connection_management.h
View File

@ -133,8 +133,9 @@ typedef struct MultiConnection
/* is the connection currently in use, and shouldn't be used by anything else */
bool claimedExclusively;
/* time connection establishment was started, for timeout */
TimestampTz connectionStart;
/* time connection establishment was started, for timeout and executor stats */
instr_time connectionEstablishmentStart;
instr_time connectionEstablishmentEnd;
/* membership in list of list of connections in ConnectionHashEntry */
dlist_node connectionNode;
@ -256,6 +257,7 @@ extern void FinishConnectionEstablishment(MultiConnection *connection);
extern void ClaimConnectionExclusively(MultiConnection *connection);
extern void UnclaimConnection(MultiConnection *connection);
extern bool IsCitusInitiatedRemoteBackend(void);
extern void MarkConnectionConnected(MultiConnection *connection);
/* time utilities */
extern double MillisecondsPassedSince(instr_time moment);