reformat

apache · Mar 19, 2024 · 1037e0a · 1037e0a
1 parent e3e0087
commit 1037e0a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/kyuubi-server/src/main/scala/org/apache/kyuubi/server/api/v1/AdminResource.scala b/kyuubi-server/src/main/scala/org/apache/kyuubi/server/api/v1/AdminResource.scala
@@ -266,9 +266,11 @@ private[v1] class AdminResource extends ApiRequestContext with Logging {
       engineNodes.foreach { node =>
         val nodePath = s"$engineSpace/$node"
         if (forceKill) {
-          val refIds = discoveryClient.getServiceNodesInfo(nodePath)
-            .flatMap(_.attributes.get("refId"))
-            .filter(StringUtils.isNotBlank(_))
+          val refIds = node.split(";")
+            .flatMap(_.split("=", 2) match {
+              case Array("refId", value) if StringUtils.isNotBlank(value) => Some(value)
+              case _ => None
+            })
 
           refIds.foreach(refId => {
             val applicationManagerInfo = ApplicationManagerInfo(