Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 18 additions & 8 deletions core/src/main/scala/kafka/migration/MigrationPropagator.scala
Original file line number Diff line number Diff line change
Expand Up @@ -70,14 +70,24 @@ class MigrationPropagator(

override def publishMetadata(image: MetadataImage): Unit = {
val oldImage = _image
val addedBrokers = new util.HashSet[Integer](image.cluster().brokers().keySet())
addedBrokers.removeAll(oldImage.cluster().brokers().keySet())
val removedBrokers = new util.HashSet[Integer](oldImage.cluster().brokers().keySet())
removedBrokers.removeAll(image.cluster().brokers().keySet())

removedBrokers.asScala.foreach(id => channelManager.removeBroker(id))
addedBrokers.asScala.foreach(id =>
channelManager.addBroker(Broker.fromBrokerRegistration(image.cluster().broker(id))))
val prevBrokers = oldImage.cluster().brokers().values().asScala
.filter(_.isMigratingZkBroker)
.filterNot(_.fenced)
.map(Broker.fromBrokerRegistration)
.toSet

val aliveBrokers = image.cluster().brokers().values().asScala
.filter(_.isMigratingZkBroker)
.filterNot(_.fenced)
.map(Broker.fromBrokerRegistration)
.toSet

val addedBrokers = aliveBrokers -- prevBrokers
val removedBrokers = prevBrokers -- aliveBrokers

stateChangeLogger.logger.debug(s"Adding brokers $addedBrokers, removing brokers $removedBrokers.")
Comment thread
mumrah marked this conversation as resolved.
Outdated
removedBrokers.foreach(broker => channelManager.removeBroker(broker.id))
addedBrokers.foreach(broker => channelManager.addBroker(broker))
_image = image
}

Expand Down