dajac commented on code in PR #12205:
URL: https://github.com/apache/kafka/pull/12205#discussion_r881435578
##########
core/src/main/scala/kafka/server/KafkaApis.scala:
##########
@@ -229,6 +229,8 @@ class KafkaApis(val requestChannel: RequestChannel,
case ApiKeys.ENVELOPE => handleEnvelope(request, requestLocal)
case ApiKeys.DESCRIBE_CLUSTER => handleDescribeCluster(request)
case ApiKeys.DESCRIBE_PRODUCERS =>
handleDescribeProducersRequest(request)
+ case ApiKeys.UNREGISTER_BROKER => maybeForwardToController(request,
Review Comment:
+1
##########
core/src/test/scala/integration/kafka/server/KRaftClusterTest.scala:
##########
@@ -725,4 +726,56 @@ class KRaftClusterTest {
cluster.close()
}
}
+
+ private def clusterImage(
+ cluster: KafkaClusterTestKit,
+ brokerId: Int
+ ): ClusterImage = {
+ cluster.brokers().get(brokerId).metadataCache.currentImage().cluster()
+ }
+
+ private def brokerIsUnfenced(
+ image: ClusterImage,
+ brokerId: Int
+ ): Boolean = {
+ Option(image.brokers().get(brokerId)) match {
+ case None => false
+ case Some(registration) => !registration.fenced()
+ }
+ }
+
+ private def brokerIsAbsent(
+ image: ClusterImage,
+ brokerId: Int
+ ): Boolean = {
+ Option(image.brokers().get(brokerId)).isEmpty
+ }
+
+ @Test
+ def testUnregisterBroker(): Unit = {
+ val cluster = new KafkaClusterTestKit.Builder(
+ new TestKitNodes.Builder().
+ setNumBrokerNodes(4).
+ setNumControllerNodes(3).build()).build()
+ try {
+ cluster.format()
+ cluster.startup()
+ cluster.waitForReadyBrokers()
+ TestUtils.waitUntilTrue(() => brokerIsUnfenced(clusterImage(cluster, 1),
0),
+ "Timed out waiting for broker 0 to be unfenced.")
+ cluster.brokers().get(0).shutdown()
+ TestUtils.waitUntilTrue(() => !brokerIsUnfenced(clusterImage(cluster,
1), 0),
+ "Timed out waiting for broker 0 to be fenced.")
+ val admin = Admin.create(cluster.clientProperties())
+ try {
Review Comment:
nit: We could use `TestUtils.resource` for auto-closing the admin and the
cluster. Not a big win so I leave it up to you.
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
