agora inbox for [email protected]help / color / mirror / Atom feed
[PATCH] chomp PQerrorMessage() in backend uses 19+ messages / 2 participants [nested] [flat]
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH] chomp PQerrorMessage() in backend uses @ 2017-02-08 14:31 Peter Eisentraut <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Peter Eisentraut @ 2017-02-08 14:31 UTC (permalink / raw) PQerrorMessage() returns an error message with a trailing newline, but in backend use (dblink, postgres_fdw, libpqwalreceiver), we want to have the error message without that for emitting via ereport(). To simplify that, add a function pchomp() that returns a pstrdup'ed string with the trailing newline characters removed. --- contrib/dblink/dblink.c | 16 ++++++------ contrib/dblink/expected/dblink.out | 2 -- contrib/postgres_fdw/connection.c | 14 ++-------- .../libpqwalreceiver/libpqwalreceiver.c | 30 +++++++++++----------- src/backend/utils/mmgr/mcxt.c | 14 ++++++++++ src/include/utils/palloc.h | 2 ++ 6 files changed, 41 insertions(+), 37 deletions(-) diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c index ac43c458cb..db0a8ba72d 100644 --- a/contrib/dblink/dblink.c +++ b/contrib/dblink/dblink.c @@ -166,7 +166,7 @@ typedef struct remoteConnHashEnt #define DBLINK_RES_INTERNALERROR(p2) \ do { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ if (res) \ PQclear(res); \ elog(ERROR, "%s: %s", p2, msg); \ @@ -204,7 +204,7 @@ typedef struct remoteConnHashEnt conn = PQconnectdb(connstr); \ if (PQstatus(conn) == CONNECTION_BAD) \ { \ - msg = pstrdup(PQerrorMessage(conn)); \ + msg = pchomp(PQerrorMessage(conn)); \ PQfinish(conn); \ ereport(ERROR, \ (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), \ @@ -278,7 +278,7 @@ dblink_connect(PG_FUNCTION_ARGS) if (PQstatus(conn) == CONNECTION_BAD) { - msg = pstrdup(PQerrorMessage(conn)); + msg = pchomp(PQerrorMessage(conn)); PQfinish(conn); if (rconn) pfree(rconn); @@ -651,7 +651,7 @@ dblink_send_query(PG_FUNCTION_ARGS) /* async query send */ retval = PQsendQuery(conn, sql); if (retval != 1) - elog(NOTICE, "could not send query: %s", PQerrorMessage(conn)); + elog(NOTICE, "could not send query: %s", pchomp(PQerrorMessage(conn))); PG_RETURN_INT32(retval); } @@ -1087,7 +1087,7 @@ storeQueryResult(volatile storeInfo *sinfo, PGconn *conn, const char *sql) PGresult *res; if (!PQsendQuery(conn, sql)) - elog(ERROR, "could not send query: %s", PQerrorMessage(conn)); + elog(ERROR, "could not send query: %s", pchomp(PQerrorMessage(conn))); if (!PQsetSingleRowMode(conn)) /* shouldn't fail */ elog(ERROR, "failed to set single-row mode for dblink query"); @@ -1366,8 +1366,8 @@ dblink_error_message(PG_FUNCTION_ARGS) DBLINK_INIT; DBLINK_GET_NAMED_CONN; - msg = PQerrorMessage(conn); - if (msg == NULL || msg[0] == '\0') + msg = pchomp(PQerrorMessage(conn)); + if (msg[0] == '\0') PG_RETURN_TEXT_P(cstring_to_text("OK")); else PG_RETURN_TEXT_P(cstring_to_text(msg)); @@ -2709,7 +2709,7 @@ dblink_res_error(PGconn *conn, const char *conname, PGresult *res, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); if (res) PQclear(res); diff --git a/contrib/dblink/expected/dblink.out b/contrib/dblink/expected/dblink.out index 5acaaf225a..4b6d26e574 100644 --- a/contrib/dblink/expected/dblink.out +++ b/contrib/dblink/expected/dblink.out @@ -377,7 +377,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect @@ -604,7 +603,6 @@ FROM dblink('myconn','SELECT * FROM foo') AS t(a int, b text, c text[]) WHERE t.a > 7; ERROR: could not establish connection DETAIL: missing "=" after "myconn" in connection info string - -- create a named persistent connection SELECT dblink_connect('myconn',connection_parameters()); dblink_connect diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index 7f7a744ac0..c6e3d44515 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -226,21 +226,11 @@ connect_pg_server(ForeignServer *server, UserMapping *user) conn = PQconnectdbParams(keywords, values, false); if (!conn || PQstatus(conn) != CONNECTION_OK) - { - char *connmessage; - int msglen; - - /* libpq typically appends a newline, strip that */ - connmessage = pstrdup(PQerrorMessage(conn)); - msglen = strlen(connmessage); - if (msglen > 0 && connmessage[msglen - 1] == '\n') - connmessage[msglen - 1] = '\0'; ereport(ERROR, (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), errmsg("could not connect to server \"%s\"", server->servername), - errdetail_internal("%s", connmessage))); - } + errdetail_internal("%s", pchomp(PQerrorMessage(conn))))); /* * Check that non-superuser has used password to establish connection; @@ -563,7 +553,7 @@ pgfdw_report_error(int elevel, PGresult *res, PGconn *conn, * return NULL, not a PGresult at all. */ if (message_primary == NULL) - message_primary = PQerrorMessage(conn); + message_primary = pchomp(PQerrorMessage(conn)); ereport(elevel, (errcode(sqlstate), diff --git a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c index 44a89c73fd..b88760fb68 100644 --- a/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c +++ b/src/backend/replication/libpqwalreceiver/libpqwalreceiver.c @@ -141,7 +141,7 @@ libpqrcv_connect(const char *conninfo, bool logical, const char *appname, conn->streamConn = PQconnectdbParams(keys, vals, /* expand_dbname = */ true); if (PQstatus(conn->streamConn) != CONNECTION_OK) { - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return NULL; } @@ -239,7 +239,7 @@ libpqrcv_identify_system(WalReceiverConn *conn, TimeLineID *primary_tli, ereport(ERROR, (errmsg("could not receive database system identifier and timeline ID from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) < 3 || PQntuples(res) != 1) { @@ -316,13 +316,13 @@ libpqrcv_startstreaming(WalReceiverConn *conn, if (!pubnames_str) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); pubnames_literal = PQescapeLiteral(conn->streamConn, pubnames_str, strlen(pubnames_str)); if (!pubnames_literal) ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); appendStringInfo(&cmd, ", publication_names %s", pubnames_literal); PQfreemem(pubnames_literal); pfree(pubnames_str); @@ -347,7 +347,7 @@ libpqrcv_startstreaming(WalReceiverConn *conn, PQclear(res); ereport(ERROR, (errmsg("could not start WAL streaming: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } PQclear(res); return true; @@ -366,7 +366,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send end-of-streaming message to primary: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); *next_tli = 0; @@ -410,7 +410,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (PQresultStatus(res) != PGRES_COMMAND_OK) ereport(ERROR, (errmsg("error reading result of streaming command: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); PQclear(res); /* Verify that there are no more results */ @@ -418,7 +418,7 @@ libpqrcv_endstreaming(WalReceiverConn *conn, TimeLineID *next_tli) if (res != NULL) ereport(ERROR, (errmsg("unexpected result after CommandComplete: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -445,7 +445,7 @@ libpqrcv_readtimelinehistoryfile(WalReceiverConn *conn, ereport(ERROR, (errmsg("could not receive timeline history file from " "the primary server: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } if (PQnfields(res) != 2 || PQntuples(res) != 1) { @@ -603,7 +603,7 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, if (PQconsumeInput(conn->streamConn) == 0) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Now that we've consumed some input, try again */ rawlen = PQgetCopyData(conn->streamConn, &conn->recvBuf, 1); @@ -630,13 +630,13 @@ libpqrcv_receive(WalReceiverConn *conn, char **buffer, PQclear(res); ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } } if (rawlen < -1) ereport(ERROR, (errmsg("could not receive data from WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); /* Return received messages to caller */ *buffer = conn->recvBuf; @@ -655,7 +655,7 @@ libpqrcv_send(WalReceiverConn *conn, const char *buffer, int nbytes) PQflush(conn->streamConn)) ereport(ERROR, (errmsg("could not send data to WAL stream: %s", - PQerrorMessage(conn->streamConn)))); + pchomp(PQerrorMessage(conn->streamConn))))); } /* @@ -689,7 +689,7 @@ libpqrcv_create_slot(WalReceiverConn *conn, const char *slotname, PQclear(res); ereport(ERROR, (errmsg("could not create replication slot \"%s\": %s", - slotname, PQerrorMessage(conn->streamConn)))); + slotname, pchomp(PQerrorMessage(conn->streamConn))))); } *lsn = DatumGetLSN(DirectFunctionCall1Coll(pg_lsn_in, InvalidOid, @@ -720,7 +720,7 @@ libpqrcv_command(WalReceiverConn *conn, const char *cmd, char **err) if (PQresultStatus(res) != PGRES_COMMAND_OK) { PQclear(res); - *err = pstrdup(PQerrorMessage(conn->streamConn)); + *err = pchomp(PQerrorMessage(conn->streamConn)); return false; } diff --git a/src/backend/utils/mmgr/mcxt.c b/src/backend/utils/mmgr/mcxt.c index 6ad0bb47b6..c21dfcb96b 100644 --- a/src/backend/utils/mmgr/mcxt.c +++ b/src/backend/utils/mmgr/mcxt.c @@ -1181,3 +1181,17 @@ pnstrdup(const char *in, Size len) out[len] = '\0'; return out; } + +/* + * Make copy of string with all trailing newline characters removed. + */ +char * +pchomp(const char *in) +{ + size_t n; + + n = strlen(in); + while (n > 0 && in[n - 1] == '\n') + n--; + return pnstrdup(in, n); +} diff --git a/src/include/utils/palloc.h b/src/include/utils/palloc.h index b72fe4aee8..2e07bd57ad 100644 --- a/src/include/utils/palloc.h +++ b/src/include/utils/palloc.h @@ -127,6 +127,8 @@ extern char *MemoryContextStrdup(MemoryContext context, const char *string); extern char *pstrdup(const char *in); extern char *pnstrdup(const char *in, Size len); +extern char *pchomp(const char *in); + /* sprintf into a palloc'd buffer --- these are in psprintf.c */ extern char *psprintf(const char *fmt,...) pg_attribute_printf(1, 2); extern size_t pvsnprintf(char *buf, size_t len, const char *fmt, va_list args) pg_attribute_printf(3, 0); -- 2.11.1 --------------8DBA76FBC0D2E1B20F0C5D8D Content-Type: text/plain Content-Disposition: inline Content-Transfer-Encoding: 8bit MIME-Version: 1.0 -- Sent via pgsql-hackers mailing list ([email protected]) To make changes to your subscription: http://www.postgresql.org/mailpref/pgsql-hackers --------------8DBA76FBC0D2E1B20F0C5D8D-- ^ permalink raw reply [nested|flat] 19+ messages in thread
* [PATCH v52 10/10] CheckLogicalDecodingRequirements: be specific about which GUC is limiting @ 2026-04-03 19:01 Álvaro Herrera <[email protected]> 0 siblings, 0 replies; 19+ messages in thread From: Álvaro Herrera @ 2026-04-03 19:01 UTC (permalink / raw) --- src/backend/commands/repack_worker.c | 3 ++- src/backend/replication/logical/logical.c | 8 +++++--- src/backend/replication/logical/logicalfuncs.c | 2 +- src/backend/replication/slot.c | 18 ++++++++++++------ src/backend/replication/slotfuncs.c | 11 ++++++----- src/backend/replication/walsender.c | 5 +++-- src/include/replication/logical.h | 3 ++- src/include/replication/slot.h | 2 +- 8 files changed, 32 insertions(+), 20 deletions(-) diff --git a/src/backend/commands/repack_worker.c b/src/backend/commands/repack_worker.c index 610592a05b0..ca827223845 100644 --- a/src/backend/commands/repack_worker.c +++ b/src/backend/commands/repack_worker.c @@ -224,7 +224,7 @@ repack_setup_logical_decoding(Oid relid) * Make sure we can use logical decoding. */ CheckSlotPermissions(); - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(true); /* * A single backend should not execute multiple REPACK commands at a time, @@ -252,6 +252,7 @@ repack_setup_logical_decoding(Oid relid) ctx = CreateInitDecodingContext(REPL_PLUGIN_NAME, NIL, true, + true, InvalidXLogRecPtr, XL_ROUTINE(.page_read = read_local_xlog_page, .segment_open = wal_segment_open, diff --git a/src/backend/replication/logical/logical.c b/src/backend/replication/logical/logical.c index f20a0fe70ad..a08aece5731 100644 --- a/src/backend/replication/logical/logical.c +++ b/src/backend/replication/logical/logical.c @@ -108,9 +108,9 @@ static void LoadOutputPlugin(OutputPluginCallbacks *callbacks, const char *plugi * decoding. */ void -CheckLogicalDecodingRequirements(void) +CheckLogicalDecodingRequirements(bool repack) { - CheckSlotRequirements(); + CheckSlotRequirements(repack); /* * NB: Adding a new requirement likely means that RestoreSlotFromDisk() @@ -305,6 +305,7 @@ StartupDecodingContext(List *output_plugin_options, * output_plugin_options -- contains options passed to the output plugin * need_full_snapshot -- if true, must obtain a snapshot able to read all * tables; if false, one that can read only catalogs is acceptable. + * for_repack -- if true, we're going to be decoding for REPACK. * restart_lsn -- if given as invalid, it's this routine's responsibility to * mark WAL as reserved by setting a convenient restart_lsn for the slot. * Otherwise, we set for decoding to start from the given LSN without @@ -325,6 +326,7 @@ LogicalDecodingContext * CreateInitDecodingContext(const char *plugin, List *output_plugin_options, bool need_full_snapshot, + bool for_repack, XLogRecPtr restart_lsn, XLogReaderRoutine *xl_routine, LogicalOutputPluginWriterPrepareWrite prepare_write, @@ -341,7 +343,7 @@ CreateInitDecodingContext(const char *plugin, * On a standby, this check is also required while creating the slot. * Check the comments in the function. */ - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(for_repack); /* shorter lines... */ slot = MyReplicationSlot; diff --git a/src/backend/replication/logical/logicalfuncs.c b/src/backend/replication/logical/logicalfuncs.c index 9760818941d..512013b0ef0 100644 --- a/src/backend/replication/logical/logicalfuncs.c +++ b/src/backend/replication/logical/logicalfuncs.c @@ -115,7 +115,7 @@ pg_logical_slot_get_changes_guts(FunctionCallInfo fcinfo, bool confirm, bool bin CheckSlotPermissions(); - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(false); if (PG_ARGISNULL(0)) ereport(ERROR, diff --git a/src/backend/replication/slot.c b/src/backend/replication/slot.c index 2c6c6773ad2..13004ed547a 100644 --- a/src/backend/replication/slot.c +++ b/src/backend/replication/slot.c @@ -1669,19 +1669,25 @@ CheckLogicalSlotExists(void) * slots. */ void -CheckSlotRequirements(void) +CheckSlotRequirements(bool repack) { + int limit; + /* * NB: Adding a new requirement likely means that RestoreSlotFromDisk() * needs the same check. */ - /* XXX we should be able to check exactly which type of slot we need */ - if (max_replication_slots + max_repack_replication_slots == 0) + if (repack) + limit = max_repack_replication_slots; + else + limit = max_replication_slots; + + if (limit == 0) ereport(ERROR, - (errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE), - errmsg("replication slots can only be used if \"%s\" > 0 or \"%s\" > 0", - "max_replication_slots", "max_repack_replication_slots"))); + errcode(ERRCODE_OBJECT_NOT_IN_PREREQUISITE_STATE), + errmsg("replication slots can only be used if \"%s\" > 0", + repack ? "max_repack_replication_slots" : "max_replication_slots")); if (wal_level < WAL_LEVEL_REPLICA) ereport(ERROR, diff --git a/src/backend/replication/slotfuncs.c b/src/backend/replication/slotfuncs.c index 78dd3c4ea66..16fbd383735 100644 --- a/src/backend/replication/slotfuncs.c +++ b/src/backend/replication/slotfuncs.c @@ -90,7 +90,7 @@ pg_create_physical_replication_slot(PG_FUNCTION_ARGS) CheckSlotPermissions(); - CheckSlotRequirements(); + CheckSlotRequirements(false); create_physical_replication_slot(NameStr(*name), immediately_reserve, @@ -164,6 +164,7 @@ create_logical_replication_slot(char *name, char *plugin, */ ctx = CreateInitDecodingContext(plugin, NIL, false, /* just catalogs is OK */ + false, /* not repack */ restart_lsn, XL_ROUTINE(.page_read = read_local_xlog_page, .segment_open = wal_segment_open, @@ -203,7 +204,7 @@ pg_create_logical_replication_slot(PG_FUNCTION_ARGS) CheckSlotPermissions(); - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(false); create_logical_replication_slot(NameStr(*name), NameStr(*plugin), @@ -240,7 +241,7 @@ pg_drop_replication_slot(PG_FUNCTION_ARGS) CheckSlotPermissions(); - CheckSlotRequirements(); + CheckSlotRequirements(false); ReplicationSlotDrop(NameStr(*name), true); @@ -648,9 +649,9 @@ copy_replication_slot(FunctionCallInfo fcinfo, bool logical_slot) CheckSlotPermissions(); if (logical_slot) - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(false); else - CheckSlotRequirements(); + CheckSlotRequirements(false); LWLockAcquire(ReplicationSlotControlLock, LW_SHARED); diff --git a/src/backend/replication/walsender.c b/src/backend/replication/walsender.c index 75ef3419a15..9d7d675fa96 100644 --- a/src/backend/replication/walsender.c +++ b/src/backend/replication/walsender.c @@ -1240,7 +1240,7 @@ CreateReplicationSlot(CreateReplicationSlotCmd *cmd) Assert(cmd->kind == REPLICATION_KIND_LOGICAL); - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(false); /* * Initially create persistent slot as ephemeral - that allows us to @@ -1309,6 +1309,7 @@ CreateReplicationSlot(CreateReplicationSlotCmd *cmd) Assert(IsLogicalDecodingEnabled()); ctx = CreateInitDecodingContext(cmd->plugin, NIL, need_full_snapshot, + false, InvalidXLogRecPtr, XL_ROUTINE(.page_read = logical_read_xlog_page, .segment_open = WalSndSegmentOpen, @@ -1466,7 +1467,7 @@ StartLogicalReplication(StartReplicationCmd *cmd) QueryCompletion qc; /* make sure that our requirements are still fulfilled */ - CheckLogicalDecodingRequirements(); + CheckLogicalDecodingRequirements(false); Assert(!MyReplicationSlot); diff --git a/src/include/replication/logical.h b/src/include/replication/logical.h index bc9d4ece672..bc075b16741 100644 --- a/src/include/replication/logical.h +++ b/src/include/replication/logical.h @@ -115,11 +115,12 @@ typedef struct LogicalDecodingContext } LogicalDecodingContext; -extern void CheckLogicalDecodingRequirements(void); +extern void CheckLogicalDecodingRequirements(bool repack); extern LogicalDecodingContext *CreateInitDecodingContext(const char *plugin, List *output_plugin_options, bool need_full_snapshot, + bool for_repack, XLogRecPtr restart_lsn, XLogReaderRoutine *xl_routine, LogicalOutputPluginWriterPrepareWrite prepare_write, diff --git a/src/include/replication/slot.h b/src/include/replication/slot.h index c316a01a807..489af7d8d6c 100644 --- a/src/include/replication/slot.h +++ b/src/include/replication/slot.h @@ -378,7 +378,7 @@ extern void ReplicationSlotDropAtPubNode(WalReceiverConn *wrconn, char *slotname extern void StartupReplicationSlots(void); extern void CheckPointReplicationSlots(bool is_shutdown); -extern void CheckSlotRequirements(void); +extern void CheckSlotRequirements(bool repack); extern void CheckSlotPermissions(void); extern ReplicationSlotInvalidationCause GetSlotInvalidationCause(const char *cause_name); -- 2.47.3 --gp2pyozrd5pweboh-- ^ permalink raw reply [nested|flat] 19+ messages in thread
end of thread, other threads:[~2026-04-03 19:01 UTC | newest] Thread overview: 19+ messages (download: mbox mbox.gz follow: Atom feed) -- links below jump to the message on this page -- 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2017-02-08 14:31 [PATCH] chomp PQerrorMessage() in backend uses Peter Eisentraut <[email protected]> 2026-04-03 19:01 [PATCH v52 10/10] CheckLogicalDecodingRequirements: be specific about which GUC is limiting Álvaro Herrera <[email protected]>
This inbox is served by agora; see mirroring instructions for how to clone and mirror all data and code used for this inbox