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

Use interrupt checking libpq wrappers in router executor.

pull/1108/head
Andres Freund 2017-01-09 14:02:45 -08:00
parent d8cdb552e5
commit c390daed0f
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/backend/distributed/executor/multi_router_executor.c
View File

@ -1168,13 +1168,12 @@ SendQueryInSingleRowMode(MultiConnection *connection, char *query,
ExtractParametersFromParamListInfo(paramListInfo, &parameterTypes, ExtractParametersFromParamListInfo(paramListInfo, &parameterTypes,
&parameterValues); &parameterValues);
querySent = PQsendQueryParams(connection->pgConn, query, querySent = SendRemoteCommandParams(connection, query, parameterCount,
parameterCount, parameterTypes, parameterValues, parameterTypes, parameterValues);
NULL, NULL, 0);
} }
else else
{ {
querySent = PQsendQuery(connection->pgConn, query); querySent = SendRemoteCommand(connection, query);
} }
if (querySent == 0) if (querySent == 0)
@ -1307,8 +1306,9 @@ StoreQueryResult(MaterialState *routerState, MultiConnection *connection,
uint32 rowCount = 0; uint32 rowCount = 0;
uint32 columnCount = 0; uint32 columnCount = 0;
ExecStatusType resultStatus = 0; ExecStatusType resultStatus = 0;
bool doRaiseInterrupts = true;
PGresult *result = PQgetResult(connection->pgConn); PGresult *result = GetRemoteCommandResult(connection, doRaiseInterrupts);
if (result == NULL) if (result == NULL)
{ {
break; break;
@ -1410,13 +1410,14 @@ ConsumeQueryResult(MultiConnection *connection, bool failOnError, int64 *rows)
*rows = 0; *rows = 0;
/* /*
* Due to single row mode we have to do multiple PQgetResult() to finish * Due to single row mode we have to do multiple GetRemoteCommandResult()
* processing of this query, even without RETURNING. For single-row mode * to finish processing of this query, even without RETURNING. For
* we have to loop until all rows are consumed. * single-row mode we have to loop until all rows are consumed.
*/ */
while (true) while (true)
{ {
PGresult *result = PQgetResult(connection->pgConn); const bool doRaiseInterrupts = true;
PGresult *result = GetRemoteCommandResult(connection, doRaiseInterrupts);
ExecStatusType status = PGRES_COMMAND_OK; ExecStatusType status = PGRES_COMMAND_OK;
if (result == NULL) if (result == NULL)