add getDatabaseLocation method by HiveCatalog

tongcheng-elong · Feb 26, 2024 · 524dc0a · 524dc0a
1 parent 52d0cdf
commit 524dc0a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/paimon-hive/paimon-hive-catalog/src/main/java/org/apache/paimon/hive/HiveCatalog.java b/paimon-hive/paimon-hive-catalog/src/main/java/org/apache/paimon/hive/HiveCatalog.java
@@ -192,6 +192,20 @@ public Path getDataTableLocation(Identifier identifier) {
             throw new RuntimeException("Can not get table " + identifier + " from metastore.", e);
         }
     }
+    public Path getDatabaseLocation(String databaseName){
+        try {
+            if (databaseExistsImpl(databaseName)) {
+                String location =
+                        locationHelper.getDatabaseLocation(client.getDatabase(databaseName));
+                if (location != null) {
+                    return new Path(location);
+                }
+            }
+        } catch (TException e) {
+            throw new RuntimeException("Can not get database " + databaseName + " from metastore.", e);
+        }
+        return null;
+    }
 
     @Override
     public List<String> listDatabases() {