Hide Hive system schemas in Hudi

trinodb · Oct 7, 2022 · 326b786 · 326b786
1 parent b44180d
commit 326b786
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/plugin/trino-hudi/src/main/java/io/trino/plugin/hudi/HudiMetadata.java b/plugin/trino-hudi/src/main/java/io/trino/plugin/hudi/HudiMetadata.java
@@ -89,7 +89,9 @@ public HudiMetadata(HiveMetastore metastore, HdfsEnvironment hdfsEnvironment, Ty
     @Override
     public List<String> listSchemaNames(ConnectorSession session)
     {
-        return metastore.getAllDatabases();
+        return metastore.getAllDatabases().stream()
+                .filter(schemaName -> !isHiveSystemSchema(schemaName))
+                .collect(toImmutableList());
     }
 
     @Override

diff --git a/plugin/trino-hudi/src/test/java/io/trino/plugin/hudi/BaseHudiConnectorTest.java b/plugin/trino-hudi/src/test/java/io/trino/plugin/hudi/BaseHudiConnectorTest.java
@@ -79,6 +79,13 @@ public void testShowCreateTable()
                         ")");
     }
 
+    @Test
+    public void testHideHiveSysSchema()
+    {
+        assertThat(computeActual("SHOW SCHEMAS").getOnlyColumnAsSet()).doesNotContain("sys");
+        assertQueryFails("SHOW TABLES IN hudi.sys", ".*Schema 'sys' does not exist");
+    }
+
     protected static String columnsToHide()
     {
         List<String> columns = new ArrayList<>(HOODIE_META_COLUMNS.size() + 1);