add topic console

This commit is contained in:
许晓东
2021-09-08 21:18:12 +08:00
parent 0b81f40b3f
commit fad17302c8
10 changed files with 196 additions and 2 deletions

View File

@@ -27,6 +27,14 @@ class KafkaConsole(config: KafkaConfig) {
}
}
protected def withAdminClientAndCatchError(f: Admin => Any, eh: Exception => Any): Any = {
try {
withAdminClient(f)
} catch {
case er: Exception => eh(er)
}
}
protected def withZKClient(f: AdminZkClient => Any): Any = {
val zkClient = KafkaZkClient(config.getZookeeperAddr, false, 30000, 30000, Int.MaxValue, Time.SYSTEM)
val adminZkClient = new AdminZkClient(zkClient)

View File

@@ -0,0 +1,37 @@
package kafka.console
import java.util
import java.util.concurrent.TimeUnit
import java.util.{Collections, List, Set}
import com.xuxd.kafka.console.config.KafkaConfig
import org.apache.kafka.clients.admin.{ListTopicsOptions, TopicDescription}
/**
* kafka-console-ui.
*
* @author xuxd
* @date 2021-09-08 19:52:27
* */
class TopicConsole(config: KafkaConfig) extends KafkaConsole(config: KafkaConfig) with Logging {
def getTopicNameList(): Set[String] = {
withAdminClientAndCatchError(admin => admin.listTopics(new ListTopicsOptions().listInternal(true)).names()
.get(3000, TimeUnit.MILLISECONDS),
e => {
log.error("listTopics error.", e)
Collections.emptySet()
}).asInstanceOf[Set[String]]
}
def getTopicList(topics: Set[String]): List[TopicDescription] = {
if (topics == null || topics.isEmpty) {
Collections.emptyList()
} else {
withAdminClientAndCatchError(admin => new util.ArrayList[TopicDescription](admin.describeTopics(topics).all().get().values()), e => {
log.error("describeTopics error.", e)
Collections.emptyList()
}).asInstanceOf[List[TopicDescription]]
}
}
}