Core: Make sqlFor case insensitive for dialect check (#9311)

apache · Dec 16, 2023 · 395e01e · 395e01e
1 parent d6a4ca7
commit 395e01e
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/core/src/main/java/org/apache/iceberg/view/BaseView.java b/core/src/main/java/org/apache/iceberg/view/BaseView.java
@@ -118,7 +118,7 @@ public SQLViewRepresentation sqlFor(String dialect) {
     for (ViewRepresentation representation : currentVersion().representations()) {
       if (representation instanceof SQLViewRepresentation) {
         SQLViewRepresentation sqlViewRepresentation = (SQLViewRepresentation) representation;
-        if (sqlViewRepresentation.dialect().equals(dialect)) {
+        if (sqlViewRepresentation.dialect().equalsIgnoreCase(dialect)) {
           return sqlViewRepresentation;
         } else if (closest == null) {
           closest = sqlViewRepresentation;

diff --git a/core/src/test/java/org/apache/iceberg/view/ViewCatalogTests.java b/core/src/test/java/org/apache/iceberg/view/ViewCatalogTests.java
@@ -1693,6 +1693,28 @@ public void testSqlForMultipleDialects() {
     assertThat(view.sqlFor("unknown-dialect").sql()).isEqualTo("select * from ns.tbl");
   }
 
+  @Test
+  public void testSqlForCaseInsensitive() {
+    TableIdentifier identifier = TableIdentifier.of("ns", "view");
+
+    if (requiresNamespaceCreate()) {
+      catalog().createNamespace(identifier.namespace());
+    }
+
+    View view =
+        catalog()
+            .buildView(identifier)
+            .withSchema(SCHEMA)
+            .withDefaultNamespace(identifier.namespace())
+            .withDefaultCatalog(catalog().name())
+            .withQuery("spark", "select * from ns.tbl")
+            .withQuery("trino", "select * from ns.tbl using X")
+            .create();
+
+    assertThat(view.sqlFor("SPARK").sql()).isEqualTo("select * from ns.tbl");
+    assertThat(view.sqlFor("TRINO").sql()).isEqualTo("select * from ns.tbl using X");
+  }
+
   @Test
   public void testSqlForInvalidArguments() {
     TableIdentifier identifier = TableIdentifier.of("ns", "view");