From 3005525d67175deec5cbb01af7565cc7690a5097 Mon Sep 17 00:00:00 2001 From: "kuroda.hayato%40jp.fujitsu.com" Date: Tue, 1 Nov 2022 09:13:42 +0000 Subject: [PATCH v24 2/3] postgres_fdw: add postgres_fdw_verify_connection_states This function can verify the status of connections that are establieshed by postgres_fdw. This check wil be done by PQConncheck(), which means this is available only on systems that support the non-standard POLLRDHUP extension to the poll system call, including Linux. This returns true if checked connection is still valid, or the checking is not supported on this platform. False is returned if the connection seems to be closed. fdw: concat string --- contrib/postgres_fdw/connection.c | 124 ++++++++++++++++++ .../postgres_fdw/postgres_fdw--1.0--1.1.sql | 15 +++ doc/src/sgml/postgres-fdw.sgml | 68 ++++++++++ 3 files changed, 207 insertions(+) diff --git a/contrib/postgres_fdw/connection.c b/contrib/postgres_fdw/connection.c index ed75ce3f79..5f345ca170 100644 --- a/contrib/postgres_fdw/connection.c +++ b/contrib/postgres_fdw/connection.c @@ -86,6 +86,9 @@ static bool xact_got_connection = false; PG_FUNCTION_INFO_V1(postgres_fdw_get_connections); PG_FUNCTION_INFO_V1(postgres_fdw_disconnect); PG_FUNCTION_INFO_V1(postgres_fdw_disconnect_all); +PG_FUNCTION_INFO_V1(postgres_fdw_verify_connection_states); +PG_FUNCTION_INFO_V1(postgres_fdw_verify_connection_states_all); +PG_FUNCTION_INFO_V1(postgres_fdw_can_verify_connection_states); /* prototypes of private functions */ static void make_new_connection(ConnCacheEntry *entry, UserMapping *user); @@ -116,6 +119,7 @@ static void pgfdw_finish_pre_subcommit_cleanup(List *pending_entries, int curlevel); static bool UserMappingPasswordRequired(UserMapping *user); static bool disconnect_cached_connections(Oid serverid); +static bool verify_cached_connections(Oid serverid); /* * Get a PGconn which can be used to execute queries on the remote PostgreSQL @@ -1862,3 +1866,123 @@ disconnect_cached_connections(Oid serverid) return result; } + +/* + * Workhorse to verify cached connections. + * + * This function scans all the connection cache entries and verifies the + * connections whose foreign server OID matches with the specified one. If + * InvalidOid is specified, it verifies all the cached connections. + * + * This function emits warnings if a disconnection is found, and this returns + * false only when the lastly verified server seems to be disconnected. + * + * Note that the verification can be used on some limited platforms. If this + * server does not support it, this function alwayse returns true. + */ +static bool +verify_cached_connections(Oid serverid) +{ + HASH_SEQ_STATUS scan; + ConnCacheEntry *entry; + bool all = !OidIsValid(serverid); + bool result = true; + StringInfoData str; + + /* quick exit if connection cache has not been initialized yet. */ + if (!ConnectionHash) + return true; + + hash_seq_init(&scan, ConnectionHash); + while ((entry = (ConnCacheEntry *) hash_seq_search(&scan))) + { + /* Ignore cache entry if no open connection right now. */ + if (!entry->conn) + continue; + + /* Skip if the entry is invalidated. */ + if (entry->invalidated) + continue; + + if (all || entry->serverid == serverid) + { + if (PQConncheck(entry->conn)) + { + /* A foreign server might be down, so construct a message. */ + ForeignServer *server = GetForeignServer(entry->serverid); + + if (result) + { + /* + * Initialize and add a prefix if this is the first + * disconnection we found. + */ + initStringInfo(&str); + appendStringInfo(&str, "could not connect to server "); + + result = false; + } + else + appendStringInfo(&str, ", "); + + appendStringInfo(&str, "\"%s\"", server->servername); + } + } + } + + /* Raise a warning if disconnections are found. */ + if (!result) + { + Assert(str.len); + ereport(WARNING, + (errcode(ERRCODE_CONNECTION_FAILURE), + errmsg("%s", str.data), + errdetail("Socket close is detected."), + errhint("Plsease check the health of server."))); + pfree(str.data); + } + + return result; +} + +/* + * Verify the specified cached connections. + * + * This function verifies the connections that are established by postgres_fdw + * from the local session to the foreign server with the given name. + * + * This function emits a warning if a disconnection is found, and this returns + * false only when the verified server seems to be disconnected. + * + * Note that the verification can be used on some limited platforms. If this + * server does not support it, this function alwayse returns true. + */ +Datum +postgres_fdw_verify_connection_states(PG_FUNCTION_ARGS) +{ + ForeignServer *server; + char *servername; + + servername = text_to_cstring(PG_GETARG_TEXT_PP(0)); + server = GetForeignServerByName(servername, false); + + PG_RETURN_BOOL(verify_cached_connections(server->serverid)); +} + +/* + * Verify all the cached connections. + * + * This function verifies all the connections that are established by postgres_fdw + * from the local session to the foreign servers. + */ +Datum +postgres_fdw_verify_connection_states_all(PG_FUNCTION_ARGS) +{ + PG_RETURN_BOOL(verify_cached_connections(InvalidOid)); +} + +Datum +postgres_fdw_can_verify_connection_states(PG_FUNCTION_ARGS) +{ + PG_RETURN_BOOL(!PQCanConncheck()); +} diff --git a/contrib/postgres_fdw/postgres_fdw--1.0--1.1.sql b/contrib/postgres_fdw/postgres_fdw--1.0--1.1.sql index ed4ca378d4..b337760bfa 100644 --- a/contrib/postgres_fdw/postgres_fdw--1.0--1.1.sql +++ b/contrib/postgres_fdw/postgres_fdw--1.0--1.1.sql @@ -18,3 +18,18 @@ CREATE FUNCTION postgres_fdw_disconnect_all () RETURNS bool AS 'MODULE_PATHNAME' LANGUAGE C STRICT PARALLEL RESTRICTED; + +CREATE FUNCTION postgres_fdw_verify_connection_states (text) +RETURNS bool +AS 'MODULE_PATHNAME' +LANGUAGE C STRICT PARALLEL RESTRICTED; + +CREATE FUNCTION postgres_fdw_verify_connection_states_all () +RETURNS bool +AS 'MODULE_PATHNAME' +LANGUAGE C STRICT PARALLEL RESTRICTED; + +CREATE FUNCTION postgres_fdw_can_verify_connection_states () +RETURNS bool +AS 'MODULE_PATHNAME' +LANGUAGE C STRICT PARALLEL SAFE; diff --git a/doc/src/sgml/postgres-fdw.sgml b/doc/src/sgml/postgres-fdw.sgml index 78f2d7d8d5..bfd1dbae93 100644 --- a/doc/src/sgml/postgres-fdw.sgml +++ b/doc/src/sgml/postgres-fdw.sgml @@ -825,6 +825,74 @@ postgres=# SELECT postgres_fdw_disconnect_all(); + + + postgres_fdw_verify_connection_states(server_name text) returns boolean + + + This function checks the status of remote connections that are established + by postgres_fdw from the local session to the foreign + server with the given name. This check is performed by polling the socket + and allows long-running transactions to be aborted sooner if the kernel + reports that the connection is closed. This function is currently + available only on systems that support the non-standard POLLRDHUP + extension to the poll system call, including Linux. This + returns true if a checked connections are still valid, + or the checking is not supported on this platform. false + is returned if the local session seems to be disconnected from other + servers. Example usage of the function: + +postgres=# SELECT postgres_fdw_verify_foreign_servers(false); + postgres_fdw_verify_connection_states +--------------------------------------- + t + + + + + + postgres_fdw_verify_connection_states_all() returns boolean + + + This function checks the status of remote connections that are established + by postgres_fdw from the local session to the foreign + servers. This check is performed by polling the socket and allows + long-running transactions to be aborted sooner if the kernel reports that + the connection is closed. This function is currently available only on + systems that support the non-standard POLLRDHUP extension + to the poll system call, including Linux. This returns + true if a lastly checked connections are still valid, + or the checking is not supported on this platform. false + is returned if the local session seems to be disconnected from other + servers. Example usage of the function: + +postgres=# SELECT postgres_fdw_verify_foreign_servers(false); + postgres_fdw_verify_connection_states_all +------------------------------------------- + t + + + + + + + postgres_fdw_can_verify_connection_states() returns boolean + + + This function checks whether functions the health of remote connectio + work well or not. This returns true if it can be used, + otherwise returns false. Example usage of the function: + + +postgres=# SELECT postgres_fdw_verify_connection_states_all(); + postgres_fdw_verify_connection_states_all +------------------------------------------- + t + + + + + -- 2.27.0