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

Add a wrapper for PQsendQuery

pull/1057/head
Marco Slot 2016-12-23 14:01:04 +01:00
parent 87c62d598e
commit 00d55ad957
2 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -223,7 +223,9 @@ ExecuteCriticalRemoteCommand(MultiConnection *connection, const char *command)
* an additional memory allocation).
*/
int
SendRemoteCommand(MultiConnection *connection, const char *command)
SendRemoteCommandParams(MultiConnection *connection, const char *command,
int parameterCount, const Oid *parameterTypes,
const char *const *parameterValues)
{
PGconn *pgConn = connection->pgConn;
bool wasNonblocking = PQisnonblocking(pgConn);
@ -237,7 +239,8 @@ SendRemoteCommand(MultiConnection *connection, const char *command)
PQsetnonblocking(pgConn, true);
}
rc = PQsendQuery(pgConn, command);
rc = PQsendQueryParams(pgConn, command, parameterCount, parameterTypes,
parameterValues, NULL, NULL, 0);
/* reset nonblocking connection to its original state */
if (!wasNonblocking)
@ -249,6 +252,19 @@ SendRemoteCommand(MultiConnection *connection, const char *command)
}
/*
* SendRemoteCommand is a PQsendQuery wrapper that logs remote commands, and
* accepts a MultiConnection instead of a plain PGconn. It makes sure it can
* send commands asynchronously without blocking (at the potential expense of
* an additional memory allocation).
*/
int
SendRemoteCommand(MultiConnection *connection, const char *command)
{
return SendRemoteCommandParams(connection, command, 0, NULL, NULL);
}
/*
* GetCommandResult is a wrapper around PQgetResult() that handles interrupts.
*

3
src/include/distributed/remote_commands.h
View File

@ -35,6 +35,9 @@ extern void LogRemoteCommand(MultiConnection *connection, const char *command);
extern void ExecuteCriticalRemoteCommand(MultiConnection *connection,
const char *command);
extern int SendRemoteCommand(MultiConnection *connection, const char *command);
extern int SendRemoteCommandParams(MultiConnection *connection, const char *command,
int parameterCount, const Oid *parameterTypes,
const char *const *parameterValues);
extern struct pg_result * GetRemoteCommandResult(MultiConnection *connection,
bool raiseInterrupts);