fix divide by zero

This commit is contained in:
Chris Lu
2024-02-04 13:38:00 -08:00
parent f9131de43f
commit 0d74ac9224

View File

@@ -69,14 +69,18 @@ func EnsureAssignmentsToActiveBrokers(activeBrokers cmap.ConcurrentMap[string,*B
}
// pick the brokers with the least number of partitions
if len(addedAssignments) > 0 {
pickedBrokers := pickBrokers(activeBrokers, int32(len(addedAssignments)))
for i, assignment := range addedAssignments {
assignment.LeaderBroker = pickedBrokers[i]
}
pickedBrokers = pickBrokers(activeBrokers, int32(len(updatedAssignments)))
}
if len(updatedAssignments) == 0 {
pickedBrokers := pickBrokers(activeBrokers, int32(len(updatedAssignments)))
for i, assignment := range updatedAssignments {
assignment.LeaderBroker = pickedBrokers[i]
}
}
return
}