lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-10-15 20:06:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix divide by zero

Browse Source
This commit is contained in:
Chris Lu
2024-02-04 13:38:00 -08:00
parent f9131de43f
commit 0d74ac9224
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
weed/mq/pub_balancer/allocate.go
View File

@@ -56,7 +56,7 @@ func pickBrokers(brokers cmap.ConcurrentMap[string, *BrokerStats], count int32)
return pickedBrokers return pickedBrokers
} }
func EnsureAssignmentsToActiveBrokers(activeBrokers cmap.ConcurrentMap[string,*BrokerStats], assignments []*mq_pb.BrokerPartitionAssignment) (addedAssignments, updatedAssignments []*mq_pb.BrokerPartitionAssignment) { func EnsureAssignmentsToActiveBrokers(activeBrokers cmap.ConcurrentMap[string, *BrokerStats], assignments []*mq_pb.BrokerPartitionAssignment) (addedAssignments, updatedAssignments []*mq_pb.BrokerPartitionAssignment) {
for _, assignment := range assignments { for _, assignment := range assignments {
if assignment.LeaderBroker == "" { if assignment.LeaderBroker == "" {
addedAssignments = append(addedAssignments, assignment) addedAssignments = append(addedAssignments, assignment)
@@ -69,13 +69,17 @@ func EnsureAssignmentsToActiveBrokers(activeBrokers cmap.ConcurrentMap[string,*B
} }
// pick the brokers with the least number of partitions // pick the brokers with the least number of partitions
pickedBrokers := pickBrokers(activeBrokers, int32(len(addedAssignments))) if len(addedAssignments) > 0 {
for i, assignment := range addedAssignments { pickedBrokers := pickBrokers(activeBrokers, int32(len(addedAssignments)))
assignment.LeaderBroker = pickedBrokers[i] for i, assignment := range addedAssignments {
assignment.LeaderBroker = pickedBrokers[i]
}
} }
pickedBrokers = pickBrokers(activeBrokers, int32(len(updatedAssignments))) if len(updatedAssignments) == 0 {
for i, assignment := range updatedAssignments { pickedBrokers := pickBrokers(activeBrokers, int32(len(updatedAssignments)))
assignment.LeaderBroker = pickedBrokers[i] for i, assignment := range updatedAssignments {
assignment.LeaderBroker = pickedBrokers[i]
}
} }
return return