get the lastOption of serverList

2020-12-21 16:31:54 +08:00 · 2020-12-21 16:31:54 +08:00 · d8e5c5aa4a
commit d8e5c5aa4a
parent e8bc44c2ad
1 changed files with 3 additions and 1 deletions
--- a/kyuubi-main/src/main/scala/org/apache/kyuubi/session/KyuubiSessionImpl.scala
+++ b/kyuubi-main/src/main/scala/org/apache/kyuubi/session/KyuubiSessionImpl.scala
@ -68,7 +68,9 @@ class KyuubiSessionImpl(
  private def getServerHost: Option[(String, Int)] = {
    try {
      val hosts = zkClient.getChildren.forPath(zkPath)
-      hosts.asScala.headOption.map { p =>
+      // TODO: use last one because to avoid touching some maybe-crashed engines
+      // We need a big improvement here.
+      hosts.asScala.lastOption.map { p =>
        val path = ZKPaths.makePath(null, zkNamespace, p)
        val hostPort = new String(zkClient.getData.forPath(path), StandardCharsets.UTF_8)
        val strings = hostPort.split(":")