[ARCTIC-115][ams] support filter database or table by keywords (#131)

【ams】support filter database or table by keywords #115
apache · Aug 25, 2022 · bb583c2 · bb583c2
1 parent 7aff61e
commit bb583c2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/ams/ams-server/src/main/java/com/netease/arctic/ams/server/controller/TableController.java b/ams/ams-server/src/main/java/com/netease/arctic/ams/server/controller/TableController.java
@@ -54,6 +54,7 @@
 import com.netease.arctic.table.TableIdentifier;
 import io.javalin.http.Context;
 import io.javalin.http.HttpCode;
+import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -344,12 +345,15 @@ public static void getTableOperations(Context ctx) {
   public static void getTableList(Context ctx) {
     String catalog = ctx.pathParam("catalog");
     String db = ctx.pathParam("db");
+    String keywords = ctx.queryParam("keywords");
 
     String thriftHost = ArcticMetaStore.conf.getString(ArcticMetaStoreConf.THRIFT_BIND_HOST);
     Integer thriftPort = ArcticMetaStore.conf.getInteger(ArcticMetaStoreConf.THRIFT_BIND_PORT);
     ArcticCatalog ac = CatalogUtil.getArcticCatalog(thriftHost, thriftPort, catalog);
     List<TableIdentifier> tableIdentifiers = ac.listTables(db);
-    List<String> tables = tableIdentifiers.stream().map(TableIdentifier::getTableName).collect(Collectors.toList());
+    List<String> tables = tableIdentifiers.stream().map(TableIdentifier::getTableName)
+            .filter(item -> StringUtils.isEmpty(keywords) || item.contains(keywords))
+            .collect(Collectors.toList());
     ctx.json(OkResponse.of(tables));
   }
 
@@ -360,11 +364,15 @@ public static void getTableList(Context ctx) {
    */
   public static void getDatabaseList(Context ctx) {
     String catalog = ctx.pathParam("catalog");
+    String keywords = ctx.queryParam("keywords");
 
     String thriftHost = ArcticMetaStore.conf.getString(ArcticMetaStoreConf.THRIFT_BIND_HOST);
     Integer thriftPort = ArcticMetaStore.conf.getInteger(ArcticMetaStoreConf.THRIFT_BIND_PORT);
     ArcticCatalog ac = CatalogUtil.getArcticCatalog(thriftHost, thriftPort, catalog);
-    ctx.json(OkResponse.of(ac.listDatabases()));
+    List<String> dbList = ac.listDatabases().stream()
+            .filter(item -> StringUtils.isEmpty(keywords) || item.contains(keywords))
+            .collect(Collectors.toList());
+    ctx.json(OkResponse.of(dbList));
   }
 
   /**