diff --git a/src/backend/distributed/commands/multi_copy.c b/src/backend/distributed/commands/multi_copy.c index aa7e4c51d..43299d922 100644 --- a/src/backend/distributed/commands/multi_copy.c +++ b/src/backend/distributed/commands/multi_copy.c @@ -1026,13 +1026,8 @@ OpenCopyTransactions(CopyStmt *copyStatement, ShardConnections *shardConnections foreach(failedPlacementCell, failedPlacementList) { ShardPlacement *failedPlacement = (ShardPlacement *) lfirst(failedPlacementCell); - uint64 shardLength = 0; - DeleteShardPlacementRow(failedPlacement->shardId, failedPlacement->nodeName, - failedPlacement->nodePort); - InsertShardPlacementRow(failedPlacement->shardId, failedPlacement->placementId, - FILE_INACTIVE, shardLength, - failedPlacement->nodeName, failedPlacement->nodePort); + UpdateShardPlacementState(failedPlacement->placementId, FILE_INACTIVE); } shardConnections->connectionList = connectionList; diff --git a/src/backend/distributed/executor/multi_router_executor.c b/src/backend/distributed/executor/multi_router_executor.c index 0eb4ad310..3056bd9f9 100644 --- a/src/backend/distributed/executor/multi_router_executor.c +++ b/src/backend/distributed/executor/multi_router_executor.c @@ -605,14 +605,8 @@ ExecuteTaskAndStoreResults(QueryDesc *queryDesc, Task *task, { ShardPlacement *failedPlacement = (ShardPlacement *) lfirst(failedPlacementCell); - uint64 shardLength = 0; - DeleteShardPlacementRow(failedPlacement->shardId, failedPlacement->nodeName, - failedPlacement->nodePort); - InsertShardPlacementRow(failedPlacement->shardId, - failedPlacement->placementId, - FILE_INACTIVE, shardLength, - failedPlacement->nodeName, failedPlacement->nodePort); + UpdateShardPlacementState(failedPlacement->placementId, FILE_INACTIVE); } executorState->es_processed = affectedTupleCount; diff --git a/src/backend/distributed/master/master_stage_protocol.c b/src/backend/distributed/master/master_stage_protocol.c index 5c0c30629..fa916c10a 100644 --- a/src/backend/distributed/master/master_stage_protocol.c +++ b/src/backend/distributed/master/master_stage_protocol.c @@ -295,12 +295,8 @@ master_append_table_to_shard(PG_FUNCTION_ARGS) uint64 placementId = placement->placementId; char *workerName = placement->nodeName; uint32 workerPort = placement->nodePort; - uint64 oldShardLength = placement->shardLength; - DeleteShardPlacementRow(shardId, workerName, workerPort); - InsertShardPlacementRow(shardId, placementId, - FILE_INACTIVE, oldShardLength, - workerName, workerPort); + UpdateShardPlacementState(placementId, FILE_INACTIVE); ereport(WARNING, (errmsg("could not append table to shard \"%s\" on node " "\"%s:%u\"", shardQualifiedName, workerName,