Mhttp://127.0.0.1:3001erge branch 'refactor/parser' of https://github.com/supabase-community/postgres_lsp into refactor/parser

Author: juleswritescode

crates/pgt_statement_splitter/src/parser.rs

@@ -134,6 +134,7 @@ impl Parser {
             .find(|t| is_relevant(t))
     }
 
+    /// Returns `None` if there are no previous relevant tokens
     fn look_back(&self) -> Option<&Token> {
         self.find_last_relevant().map(|it| it.1)
     }

crates/pgt_statement_splitter/src/parser/common.rs

@@ -145,6 +145,12 @@ pub(crate) fn unknown(p: &mut Parser, exclude: &[SyntaxKind]) {
                         SyntaxKind::Also,
                         // for create rule
                         SyntaxKind::Instead,
+                        // for UNION
+                        SyntaxKind::Union,
+                        // for UNION ALL
+                        SyntaxKind::All,
+                        // for UNION ... EXCEPT
+                        SyntaxKind::Except,
                     ]
                     .iter()
                     .all(|x| Some(x) != prev.as_ref())

crates/pgt_statement_splitter/tests/data/simple_union__4.sql

@@ -0,0 +1,7 @@
+select 1 union all select 2;
+
+select 1 union select 2;
+
+select 1 union select 2 except select 3;
+
+select 1 union all select 2 except select 3;