Add handling for ALTER statements with BEGIN in SQL parser

Signed-off-by: worksofliam <[email protected]>

src/language/sql/document.ts

@@ -87,6 +87,10 @@ export default class Document {
                   break;
                 }
 
+                if (upperValue === `BEGIN` && currentStatementType === StatementType.Alter) {
+                  break;
+                }
+
                 // We include BEGIN in the current statement
                 // then the next statement beings
                 const statementTokens = tokens.slice(statementStart, i+1);

src/language/sql/tests/blocks.test.ts

@@ -539,4 +539,19 @@ test('CREATE statements', () => {
 
   const groups = doc.getStatementGroups();
   expect(groups.length).toBe(3);
-});
+});
+
+test(`ALTER with BEGIN`, () => {
+  const lines = [
+    `ALTER TABLE mylongtable1`,
+    `  ADD COLUMN ib TIMESTAMP(12) NOT NULL GENERATED ALWAYS AS ROW BEGIN`,
+    `  ADD COLUMN ie TIMESTAMP(12) NOT NULL GENERATED ALWAYS AS ROW END`,
+    `  ADD COLUMN tid TIMESTAMP(12) NOT NULL GENERATED ALWAYS AS TRANSACTION START`,
+    `  ID`,
+    `  ADD PERIOD SYSTEM_TIME (ib, ie);`,
+  ].join(`\n`);
+
+  const doc = new Document(lines);
+  const groups = doc.getStatementGroups();
+  expect(groups.length).toBe(1);
+})