diff --git a/kyuubi-ctl/src/test/scala/org/apache/kyuubi/ctl/ControlCliSuite.scala b/kyuubi-ctl/src/test/scala/org/apache/kyuubi/ctl/ControlCliSuite.scala index 43a694a08..b0319e497 100644 --- a/kyuubi-ctl/src/test/scala/org/apache/kyuubi/ctl/ControlCliSuite.scala +++ b/kyuubi-ctl/src/test/scala/org/apache/kyuubi/ctl/ControlCliSuite.scala @@ -190,9 +190,9 @@ class ControlCliSuite extends KyuubiFunSuite with TestPrematureExit { assert(children.size == 2) assert(children.head.startsWith( - s"serviceUri=localhost:10000;version=$KYUUBI_VERSION;sequence=")) + s"serverUri=localhost:10000;version=$KYUUBI_VERSION;sequence=")) assert(children.last.startsWith( - s"serviceUri=localhost:10001;version=$KYUUBI_VERSION;sequence=")) + s"serverUri=localhost:10001;version=$KYUUBI_VERSION;sequence=")) children.foreach { child => framework.delete(s"""$znodeRoot/$child""") } diff --git a/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/etcd/EtcdDiscoveryClient.scala b/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/etcd/EtcdDiscoveryClient.scala index d979804f4..7edc7e8a3 100644 --- a/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/etcd/EtcdDiscoveryClient.scala +++ b/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/etcd/EtcdDiscoveryClient.scala @@ -335,7 +335,7 @@ class EtcdDiscoveryClient(conf: KyuubiConf) extends DiscoveryClient { val extraInfo = attributes.map(kv => kv._1 + "=" + kv._2).mkString(";", ";", "") val pathPrefix = DiscoveryPaths.makePath( namespace, - s"serviceUri=$instance;version=${version.getOrElse(KYUUBI_VERSION)}" + + s"serverUri=$instance;version=${version.getOrElse(KYUUBI_VERSION)}" + s"${extraInfo.stripSuffix(";")};${session}sequence=") val znode = instance diff --git a/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClient.scala b/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClient.scala index 2db7d89d6..a06087d3a 100644 --- a/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClient.scala +++ b/kyuubi-ha/src/main/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClient.scala @@ -361,7 +361,7 @@ class ZookeeperDiscoveryClient(conf: KyuubiConf) extends DiscoveryClient { val extraInfo = attributes.map(kv => kv._1 + "=" + kv._2).mkString(";", ";", "") val pathPrefix = ZKPaths.makePath( namespace, - s"serviceUri=$instance;version=${version.getOrElse(KYUUBI_VERSION)}" + + s"serverUri=$instance;version=${version.getOrElse(KYUUBI_VERSION)}" + s"${extraInfo.stripSuffix(";")};${session}sequence=") var localServiceNode: PersistentNode = null val createMode = diff --git a/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/DiscoveryClientTests.scala b/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/DiscoveryClientTests.scala index 9caf38646..53c0586f5 100644 --- a/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/DiscoveryClientTests.scala +++ b/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/DiscoveryClientTests.scala @@ -60,7 +60,7 @@ trait DiscoveryClientTests extends KyuubiFunSuite { assert(discoveryClient.pathExists(basePath)) val children = discoveryClient.getChildren(basePath) assert(children.head === - s"serviceUri=${service.frontendServices.head.connectionUrl};" + + s"serverUri=${service.frontendServices.head.connectionUrl};" + s"version=$KYUUBI_VERSION;sequence=0000000000") children.foreach { child => @@ -107,7 +107,7 @@ trait DiscoveryClientTests extends KyuubiFunSuite { assert(discoveryClient.pathExists(basePath)) val children = discoveryClient.getChildren(basePath) assert(children.head === - s"serviceUri=${service.frontendServices.head.connectionUrl};" + + s"serverUri=${service.frontendServices.head.connectionUrl};" + s"version=$KYUUBI_VERSION;sequence=0000000000") children.foreach { child => diff --git a/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClientSuite.scala b/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClientSuite.scala index dd78e1fb8..34ed05593 100644 --- a/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClientSuite.scala +++ b/kyuubi-ha/src/test/scala/org/apache/kyuubi/ha/client/zookeeper/ZookeeperDiscoveryClientSuite.scala @@ -196,7 +196,7 @@ abstract class ZookeeperDiscoveryClientSuite extends DiscoveryClientTests assert(discoveryClient.pathExists(basePath)) val children = discoveryClient.getChildren(basePath) assert(children.head === - s"serviceUri=${service.frontendServices.head.connectionUrl};" + + s"serverUri=${service.frontendServices.head.connectionUrl};" + s"version=$KYUUBI_VERSION;sequence=0000000000") children.foreach { child => diff --git a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/AdminResourceSuite.scala b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/AdminResourceSuite.scala index 8c5ccc3be..0951d8272 100644 --- a/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/AdminResourceSuite.scala +++ b/kyuubi-server/src/test/scala/org/apache/kyuubi/server/api/v1/AdminResourceSuite.scala @@ -695,8 +695,8 @@ class AdminResourceSuite extends KyuubiFunSuite with RestFrontendTestHelper { assert(restFrontendService.connectionUrl.equals(testServer.getInstance())) assert(!testServer.getAttributes.isEmpty) val attributes = testServer.getAttributes - assert(attributes.containsKey("serviceUri") && - attributes.get("serviceUri").equals(fe.connectionUrl)) + assert(attributes.containsKey("serverUri") && + attributes.get("serverUri").equals(fe.connectionUrl)) assert(attributes.containsKey("version")) assert(attributes.containsKey("sequence")) assert("Running".equals(testServer.getStatus))