Support showDatabase method and etc (#209)

* Support showDatabase method * Minor Fix * Support the id field in TDDatabase again
treasure-data · Jan 19, 2023 · ba7182d · ba7182d
1 parent 9049c14
commit ba7182d
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 0 deletions.
diff --git a/src/main/java/com/treasuredata/client/TDClient.java b/src/main/java/com/treasuredata/client/TDClient.java
@@ -364,6 +364,13 @@ public List<TDDatabase> listDatabases()
         return doGet("/v3/database/list", new TypeReference<List<TDDatabase>>() {});
     }
 
+    @Override
+    public TDDatabase showDatabase(String databaseName)
+            throws TDClientException
+    {
+        return doGet(buildUrl("/v3/database/show", databaseName), TDDatabase.class);
+    }
+
     private static Pattern acceptableNamePattern = Pattern.compile("^([a-z0-9_]+)$");
 
     static String validateDatabaseName(String databaseName)

diff --git a/src/main/java/com/treasuredata/client/TDClientApi.java b/src/main/java/com/treasuredata/client/TDClientApi.java
@@ -116,6 +116,14 @@ public interface TDClientApi<ClientImpl>
      */
     List<String> listDatabaseNames();
 
+    /**
+     * Get the detailed information of a database
+     *
+     * @return TDDatabase
+     * @throws TDClientException if failed to retrieve the detailed information of a database.
+     */
+    TDDatabase showDatabase(String databaseName);
+
     /**
      * Get the detailed information of databases
      *

diff --git a/src/main/java/com/treasuredata/client/model/TDDatabase.java b/src/main/java/com/treasuredata/client/model/TDDatabase.java
@@ -26,6 +26,7 @@
 @JsonCollectionRootName(value = "databases")
 public class TDDatabase
 {
+    private final String id;
     private final String name;
     private final long count;
     private final String createdAt;
@@ -35,6 +36,7 @@ public class TDDatabase
 
     @JsonCreator
     public TDDatabase(
+            @JsonProperty("id") String id,
             @JsonProperty("name") String name,
             @JsonProperty("count") long count,
             @JsonProperty("created_at") String createdAt,
@@ -43,6 +45,7 @@ public TDDatabase(
             @JsonProperty("permission") String permission
     )
     {
+        this.id = id;
         this.name = name;
         this.count = count;
         this.createdAt = createdAt;
@@ -51,6 +54,12 @@ public TDDatabase(
         this.permission = permission;
     }
 
+    @JsonProperty
+    public String getId()
+    {
+        return id;
+    }
+
     @JsonProperty
     public String getName()
     {

diff --git a/src/test/java/com/treasuredata/client/TestTDClient.java b/src/test/java/com/treasuredata/client/TestTDClient.java
@@ -207,6 +207,18 @@ public void serverStatus()
         assertEquals("ok", s.getString("status"));
     }
 
+    @Test
+    public void showDatabase()
+            throws Exception
+    {
+        String databaseName = "sample_datasets";
+        TDDatabase dbDetail = client.showDatabase(databaseName);
+        assertEquals("should match in sample_datasets", databaseName, dbDetail.getName());
+        assertTrue("should be positive", Integer.parseInt(dbDetail.getId()) > 0);
+
+        logger.debug(dbDetail.toString());
+    }
+
     @Test
     public void listDatabases()
             throws Exception