gsh-digital-services/seaweedFS
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

erasure coding: add cleanup step if anything goes wrong

Browse Source
This commit is contained in:
Chris Lu
2021-09-13 01:55:49 -07:00
parent f74b29416a
commit 6cd1ce8b74
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
weed/shell/command_ec_encode.go
View File

@@ -203,6 +203,14 @@ func parallelCopyEcShardsFromSource(grpcDialOption grpc.DialOption, targetServer
server.addEcVolumeShards(volumeId, collection, copiedShardIds) server.addEcVolumeShards(volumeId, collection, copiedShardIds)
} }
} }
cleanupFunc := func(server *EcNode, allocatedEcShardIds []uint32) {
if err := unmountEcShards(grpcDialOption, volumeId, pb.NewServerAddressFromDataNode(server.info), allocatedEcShardIds); err != nil {
fmt.Printf("unmount aborted shards %d.%v on %s: %v\n", volumeId, allocatedEcShardIds, server.info.Id, err)
}
if err := sourceServerDeleteEcShards(grpcDialOption, collection, volumeId, pb.NewServerAddressFromDataNode(server.info), allocatedEcShardIds); err != nil {
fmt.Printf("remove aborted shards %d.%v on %s: %v\n", volumeId, allocatedEcShardIds, server.info.Id, err)
}
}
// maybe parallelize // maybe parallelize
for i, server := range targetServers { for i, server := range targetServers {
@@ -221,6 +229,12 @@ func parallelCopyEcShardsFromSource(grpcDialOption grpc.DialOption, targetServer
close(shardIdChan) close(shardIdChan)
if err != nil { if err != nil {
for i, server := range targetServers {
if len(allocatedEcIds[i]) <= 0 {
continue
}
cleanupFunc(server, allocatedEcIds[i])
}
return nil, err return nil, err
} }