Fix/split listener (#228)

* feat: improve FlinkSqlSplitListener

* feat: improve ImpalaSqlSplitListener

* feat: improve MysqlSplitListener

* fix: correct PgSqlSplitListener

* feat: improve TrinoSqlSplitListener

* test: add split listener unit test

* chore: ignore iml file

* feat: add pgsql missing rules

* test: fix pgsql unit tests

test/parser/flinksql/listener.test.ts

@@ -22,4 +22,46 @@ describe('Flink SQL Listener Tests', () => {
         await parser.listen(listenTableName as ParseTreeListener, parseTree);
         expect(result).toBe(expectTableName);
     });
+
+    test('Split sql listener', async () => {
+        const singleStatementArr = [
+            `SELECT id FROM games ORDER BY score;`,
+
+            `INSERT INTO country_page_view
+        SELECT user1, cnt FROM page_view_source`,
+
+            `BEGIN STATEMENT SET;
+            INSERT INTO country_page_view
+            VALUES ('Chinese', 'mumiao', 18),
+                ('Amercian', 'georage', 22);
+            INSERT INTO country_page_view
+            VALUES ('Chinese', 'mumiao', 18),
+                ('Amercian', 'georage', 22);
+            END;`,
+
+            `;`,
+        ];
+        const sql = singleStatementArr.join('\n');
+        const sqlSlices = parser.splitSQLByStatement(sql);
+
+        expect(sqlSlices).not.toBeNull();
+
+        // check text in result
+        expect(sqlSlices.map((item) => item.text)).toEqual(singleStatementArr);
+
+        // check startIndex and endIndex in result
+        sqlSlices.forEach((slice, index) => {
+            expect(sql.slice(slice.startIndex, slice.endIndex + 1)).toBe(singleStatementArr[index]);
+        });
+
+        // check lineNumber in result
+        expect(sqlSlices[0].startLine).toBe(1);
+        expect(sqlSlices[0].endLine).toBe(1);
+        expect(sqlSlices[1].startLine).toBe(2);
+        expect(sqlSlices[1].endLine).toBe(3);
+        expect(sqlSlices[2].startLine).toBe(4);
+        expect(sqlSlices[2].endLine).toBe(11);
+        expect(sqlSlices[3].startLine).toBe(12);
+        expect(sqlSlices[3].endLine).toBe(12);
+    });
 });