Snowflake: parse (EXECUTE IMMEDIATE) as RESULTSET/CURSOR payload

Author: sabir-akhadov-localstack

src/ast/query.rs

@@ -180,6 +180,11 @@ pub enum SetExpr {
     /// `SHOW` statement used as the query payload of a Snowflake `RESULTSET` or
     /// `CURSOR` declaration (`RESULTSET DEFAULT (SHOW ...)`, `CURSOR FOR (SHOW ...)`).
     Show(Statement),
+    /// `EXECUTE IMMEDIATE` statement used as the query payload of a Snowflake
+    /// `RESULTSET` declaration (`RESULTSET := (EXECUTE IMMEDIATE '<query>')`). The
+    /// payload SQL is only known at runtime, so the contained statement carries
+    /// the dynamic-SQL expression rather than a static query body.
+    Execute(Statement),
 }
 
 impl SetExpr {
@@ -209,6 +214,7 @@ impl fmt::Display for SetExpr {
             SetExpr::Merge(v) => v.fmt(f),
             SetExpr::Table(t) => t.fmt(f),
             SetExpr::Show(v) => v.fmt(f),
+            SetExpr::Execute(v) => v.fmt(f),
             SetExpr::SetOperation {
                 left,
                 right,

src/ast/spans.rs

@@ -236,6 +236,7 @@ impl Spanned for SetExpr {
             SetExpr::Delete(statement) => statement.span(),
             SetExpr::Merge(statement) => statement.span(),
             SetExpr::Show(statement) => statement.span(),
+            SetExpr::Execute(statement) => statement.span(),
         }
     }
 }

src/parser/mod.rs

@@ -8361,6 +8361,30 @@ impl<'a> Parser<'a> {
     /// [`SetExpr::Show`]. The pre-existing `(SELECT ...)` subquery path is left
     /// untouched.
     fn parse_snowflake_declaration_payload_expr(&mut self) -> Result<Expr, ParserError> {
+        let is_paren_execute = self.dialect.supports_execute_immediate()
+            && self.peek_nth_token_ref(0).token == Token::LParen
+            && matches!(
+                &self.peek_nth_token_ref(1).token,
+                Token::Word(w) if w.keyword == Keyword::EXECUTE
+            );
+        if is_paren_execute {
+            self.expect_token(&Token::LParen)?;
+            self.expect_keyword_is(Keyword::EXECUTE)?;
+            let execute = self.parse_execute()?;
+            self.expect_token(&Token::RParen)?;
+            return Ok(Expr::Subquery(Box::new(Query {
+                with: None,
+                body: Box::new(SetExpr::Execute(execute)),
+                order_by: None,
+                limit_clause: None,
+                fetch: None,
+                locks: vec![],
+                for_clause: None,
+                settings: None,
+                format_clause: None,
+                pipe_operators: vec![],
+            })));
+        }
         let is_paren_show = self.dialect.supports_show_in_resultset_cursor()
             && self.peek_nth_token_ref(0).token == Token::LParen
             && matches!(