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

Check connection status before using it

pull/1762/head
metdos 2017-11-06 13:53:23 +02:00
parent 2332678793
commit c83edc36b5
4 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -424,7 +424,7 @@ SendRemoteCommandParams(MultiConnection *connection, const char *command,
* Don't try to send command if connection is entirely gone
* (PQisnonblocking() would crash).
*/
if (!pgConn)
if (!pgConn || PQstatus(pgConn) != CONNECTION_OK)
{
return 0;
}
@ -457,7 +457,7 @@ SendRemoteCommand(MultiConnection *connection, const char *command)
* Don't try to send command if connection is entirely gone
* (PQisnonblocking() would crash).
*/
if (!pgConn)
if (!pgConn || PQstatus(pgConn) != CONNECTION_OK)
{
return 0;
}

6
src/test/regress/expected/multi_replicate_reference_table.out
View File

@ -713,6 +713,12 @@ SELECT 1 FROM master_add_node('localhost', :worker_2_port);
1
(1 row)
-- test adding an invalid node while we have reference tables to replicate
-- set client message level to ERROR to suppress OS-dependent host name resolution warnings
SET client_min_messages to ERROR;
SELECT master_add_node('invalid-node-name', 9999);
ERROR: failure on connection marked as essential: invalid-node-name:9999
SET client_min_messages to DEFAULT;
-- drop unnecassary tables
DROP TABLE initially_not_replicated_reference_table;
-- reload pg_dist_shard_placement table

4
src/test/regress/expected/multi_router_planner.out
View File

@ -2151,7 +2151,7 @@ GRANT INSERT ON ALL TABLES IN SCHEMA public TO router_user;
BEGIN;
INSERT INTO failure_test VALUES (1, 1);
WARNING: connection error: localhost:57638
DETAIL: no connection to the server
DETAIL: FATAL: role "router_user" does not exist
SELECT shardid, shardstate, nodename, nodeport FROM pg_dist_shard_placement
WHERE shardid IN (
SELECT shardid FROM pg_dist_shard
@ -2169,7 +2169,7 @@ SELECT shardid, shardstate, nodename, nodeport FROM pg_dist_shard_placement
ROLLBACK;
INSERT INTO failure_test VALUES (2, 1);
WARNING: connection error: localhost:57638
DETAIL: no connection to the server
DETAIL: FATAL: role "router_user" does not exist
SELECT shardid, shardstate, nodename, nodeport FROM pg_dist_shard_placement
WHERE shardid IN (
SELECT shardid FROM pg_dist_shard

7
src/test/regress/sql/multi_replicate_reference_table.sql
View File

@ -457,6 +457,13 @@ ORDER BY 1,4,5;
-- this should have no effect
SELECT 1 FROM master_add_node('localhost', :worker_2_port);
-- test adding an invalid node while we have reference tables to replicate
-- set client message level to ERROR to suppress OS-dependent host name resolution warnings
SET client_min_messages to ERROR;
SELECT master_add_node('invalid-node-name', 9999);
SET client_min_messages to DEFAULT;
-- drop unnecassary tables
DROP TABLE initially_not_replicated_reference_table;