diff --git a/src/backend/distributed/commands/multi_copy.c b/src/backend/distributed/commands/multi_copy.c index f681f7801..44442c0ab 100644 --- a/src/backend/distributed/commands/multi_copy.c +++ b/src/backend/distributed/commands/multi_copy.c @@ -3585,6 +3585,12 @@ CopyGetPlacementConnection(HTAB *connectionStateHash, ShardPlacement *placement, */ Assert(connection != NULL); + /* + * Make sure that the connection management remembers that Citus + * accesses this placement over the connection. + */ + AssignPlacementListToConnection(list_make1(placementAccess), connection); + return connection; }