feat: split catalog and schema

Signed-off-by: Andreas Reichel <[email protected]>

Author: manticore-projects

src/main/java/net/sf/jsqlparser/schema/Table.java

@@ -170,7 +170,44 @@ public String getUnquotedSchemaName() {
     }
 
     public Table setSchemaName(String schemaName) {
-        this.setIndex(SCHEMA_IDX, schemaName);
+
+        // BigQuery seems to allow things like: `catalogName.schemaName.tableName` in only one pair
+        // of quotes
+        // however, some people believe that Dots in Names are a good idea, so provide a switch-off
+        boolean splitNamesOnDelimiter = System.getProperty("SPLIT_NAMES_ON_DELIMITER") == null ||
+                !List
+                        .of("0", "N", "n", "FALSE", "false", "OFF", "off")
+                        .contains(System.getProperty("SPLIT_NAMES_ON_DELIMITER"));
+
+        if (MultiPartName.isQuoted(schemaName) && schemaName.contains(".") && splitNamesOnDelimiter) {
+            String[] parts = MultiPartName.unquote(schemaName).split("\\.");
+            switch (parts.length) {
+                case 2:
+                    setIndex(DATABASE_IDX, "\"" + parts[0] + "\"");
+                    setIndex(SCHEMA_IDX, "\"" + parts[1] + "\"");
+                    break;
+                case 1:
+                    setIndex(SCHEMA_IDX, "\"" + parts[0] + "\"");
+                    break;
+                default:
+                    throw new RuntimeException("Invalid schema name: " + schemaName);
+            }
+        } else if (schemaName.contains(".") && splitNamesOnDelimiter) {
+            String[] parts = MultiPartName.unquote(schemaName).split("\\.");
+            switch (parts.length) {
+                case 2:
+                    setIndex(DATABASE_IDX, parts[0]);
+                    setIndex(SCHEMA_IDX, parts[1]);
+                    break;
+                case 1:
+                    setIndex(SCHEMA_IDX, parts[0]);
+                    break;
+                default:
+                    throw new RuntimeException("Invalid schema name: " + schemaName);
+            }
+        } else {
+            this.setIndex(SCHEMA_IDX, schemaName);
+        }
         return this;
     }
 

src/test/java/net/sf/jsqlparser/schema/TableTest.java

@@ -120,4 +120,13 @@ void testClone() {
         Assertions.assertNotSame(t.clone(), t);
         Assertions.assertNotEquals(t.clone(), t);
     }
+
+    @Test
+    void testWithSchema() {
+        Table t = new Table("a");
+        t.setSchemaName("UNNAMED.session1");
+
+        Assertions.assertEquals("UNNAMED", t.getDatabaseName());
+        Assertions.assertEquals("session1", t.getSchemaName());
+    }
 }

src/test/java/net/sf/jsqlparser/statement/create/CreateTableTest.java

@@ -1073,4 +1073,15 @@ void testUniqueAfterForeignKeyIssue2082() throws JSQLParserException {
                         ", UNIQUE (employee_name));";
         assertSqlCanBeParsedAndDeparsed(sqlStr, true);
     }
+
+    @Test
+    void testWithCatalog() throws JSQLParserException {
+        String sqlStr="CREATE TABLE UNNAMED.session1.a (b VARCHAR (1))";
+        CreateTable st = (CreateTable) assertSqlCanBeParsedAndDeparsed(sqlStr, true);
+
+        Table t = st.getTable();
+        assertEquals("UNNAMED", t.getCatalogName());
+        assertEquals("session1", t.getSchemaName());
+        assertEquals("a", t.getUnquotedName());
+    }
 }