ramesh-muthusamy commented on a change in pull request #9319: URL: https://github.com/apache/kafka/pull/9319#discussion_r512938623
########## File path: connect/runtime/src/main/java/org/apache/kafka/connect/runtime/distributed/IncrementalCooperativeAssignor.java ########## @@ -445,16 +444,34 @@ protected void handleLostAssignments(ConnectorsAndTasks lostAssignments, if (scheduledRebalance > 0 && now >= scheduledRebalance) { // delayed rebalance expired and it's time to assign resources log.debug("Delayed rebalance expired. Reassigning lost tasks"); - Optional<WorkerLoad> candidateWorkerLoad = Optional.empty(); + List<WorkerLoad> candidateWorkerLoad = Collections.emptyList(); if (!candidateWorkersForReassignment.isEmpty()) { candidateWorkerLoad = pickCandidateWorkerForReassignment(completeWorkerAssignment); } - if (candidateWorkerLoad.isPresent()) { - WorkerLoad workerLoad = candidateWorkerLoad.get(); - log.debug("A candidate worker has been found to assign lost tasks: {}", workerLoad.worker()); - lostAssignments.connectors().forEach(workerLoad::assign); - lostAssignments.tasks().forEach(workerLoad::assign); + if (!candidateWorkerLoad.isEmpty()) { + log.debug("Assigning lost tasks to {} candidate workers: {}", + candidateWorkerLoad.size(), + candidateWorkerLoad.stream().map(WorkerLoad::worker).collect(Collectors.joining(","))); + Iterator<WorkerLoad> candidateWorkerIterator = candidateWorkerLoad.iterator(); + for (String connector : lostAssignments.connectors()) { + // Loop over the the candidate workers as many times as it takes Review comment: It not only balances the tasks for new workers it also does revocation if there is any overloaded worker. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org