api

apache · Feb 4, 2024 · ce017b8 · ce017b8
1 parent 576379c
commit ce017b8
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/kyuubi-rest-client/src/main/java/org/apache/kyuubi/client/AdminRestApi.java b/kyuubi-rest-client/src/main/java/org/apache/kyuubi/client/AdminRestApi.java
@@ -18,6 +18,7 @@
 package org.apache.kyuubi.client;
 
 import java.util.*;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.kyuubi.client.api.v1.dto.Engine;
 import org.apache.kyuubi.client.api.v1.dto.OperationData;
 import org.apache.kyuubi.client.api.v1.dto.ServerData;
@@ -103,10 +104,24 @@ public String closeSession(String sessionHandleStr) {
   }
 
   public List<OperationData> listOperations() {
+    return listOperations(Collections.emptyList(), null);
+  }
+
+  public List<OperationData> listOperations(List<String> users, String sessionHandleStr) {
+    Map<String, Object> params = new HashMap<>();
+    if (users != null && !users.isEmpty()) {
+      params.put("users", String.join(",", users));
+    }
+    if (StringUtils.isNotBlank(sessionHandleStr)) {
+      params.put("sessionHandle", sessionHandleStr);
+    }
     OperationData[] result =
         this.getClient()
             .get(
-                API_BASE_PATH + "/operations", null, OperationData[].class, client.getAuthHeader());
+                API_BASE_PATH + "/operations",
+                params,
+                OperationData[].class,
+                client.getAuthHeader());
     return Arrays.asList(result);
   }