root
/
kafka
mirror of https://github.com/apache/kafka.git
1
0
Fork 0
Code Issues Actions 2 Packages Projects Releases Wiki Activity

KAFKA-18412: Remove EmbeddedZookeeper (#18399) 

Reviewers: Mickael Maison <mickael.maison@gmail.com>
This commit is contained in:
TengYao Chi 2025-01-08 00:08:00 +08:00 committed by GitHub
parent 2fc35c81be
commit af255a0c37
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 0 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
checkstyle/import-control.xml
View File

@ -425,7 +425,6 @@
<allow pkg="kafka.utils" /> <allow pkg="kafka.utils" />
<allow pkg="kafka.log" /> <allow pkg="kafka.log" />
<allow pkg="scala" /> <allow pkg="scala" />
<allow class="kafka.zk.EmbeddedZookeeper"/>
<allow pkg="com.fasterxml.jackson" /> <allow pkg="com.fasterxml.jackson" />
<allow pkg="org.apache.kafka.tools" /> <allow pkg="org.apache.kafka.tools" />
<allow pkg="org.apache.kafka.server.config" /> <allow pkg="org.apache.kafka.server.config" />

79
core/src/test/scala/unit/kafka/zk/EmbeddedZookeeper.scala
View File

@ -1,79 +0,0 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package kafka.zk
import org.apache.zookeeper.server.{NIOServerCnxnFactory, SessionTrackerImpl, ZooKeeperServer}
import kafka.utils.{CoreUtils, Logging, TestUtils}
import java.net.InetSocketAddress
import org.apache.kafka.common.utils.Utils
import java.io.Closeable
/**
* ZooKeeperServer wrapper that starts the server with temporary directories during construction and deletes
* the directories when `shutdown()` is called.
*
* This is an internal class and it's subject to change. We recommend that you implement your own simple wrapper
* if you need similar functionality.
*/
// This should be named EmbeddedZooKeeper for consistency with other classes, but since this is widely used by other
// projects (even though it's internal), we keep the name as it is until we have a publicly supported test library for
// others to use.
class EmbeddedZookeeper extends Closeable with Logging {
val snapshotDir = TestUtils.tempDir()
val logDir = TestUtils.tempDir()
val tickTime = 800 // allow a maxSessionTimeout of 20 * 800ms = 16 secs
System.setProperty("zookeeper.forceSync", "no") //disable fsync to ZK txn log in tests to avoid timeout
val zookeeper = new ZooKeeperServer(snapshotDir, logDir, tickTime)
val factory = new NIOServerCnxnFactory()
private val addr = new InetSocketAddress("127.0.0.1", TestUtils.RandomPort)
factory.configure(addr, 0)
factory.startup(zookeeper)
val port = zookeeper.getClientPort
def shutdown(): Unit = {
// Also shuts down ZooKeeperServer
CoreUtils.swallow(factory.shutdown(), this)
zookeeper.getSessionTracker match {
case tracker: SessionTrackerImpl =>
while (tracker.isAlive) {
Thread.sleep(100)
}
case _ =>
}
def isDown(): Boolean = {
try {
ZkFourLetterWords.sendStat("127.0.0.1", port, 3000)
false
} catch { case _: Throwable => true }
}
Iterator.continually(isDown()).exists(identity)
CoreUtils.swallow(zookeeper.getZKDatabase.close(), this)
Utils.delete(logDir)
Utils.delete(snapshotDir)
}
override def close(): Unit = shutdown()
}