LAV-2586: Support CREATE TABLE USING TEMPLATE (#2984)

* LAV-2586: support CREATE TABLE USING TEMPLATE

Co-authored-by: Sabir Akhadov <sabir.akhadov@localstack.cloud>

* LAV-2586: update vendored CreateTable fixtures

Co-authored-by: Sabir Akhadov <sabir.akhadov@localstack.cloud>

* LAV-2586: close template validation gaps

Co-authored-by: Sabir Akhadov <sabir.akhadov@localstack.cloud>

---------

Co-authored-by: spiral <spiral@localhost>
Co-authored-by: Sabir Akhadov <sabir.akhadov@localstack.cloud>

Author: 3 people

src/ast/ddl.rs

@@ -3852,6 +3852,8 @@ pub struct CreateTable {
     pub location: Option<String>,
     /// Query used to populate the table
     pub query: Option<Box<Query>>,
+    /// Snowflake `USING TEMPLATE (<query>)` query.
+    pub template: Option<Box<Query>>,
     /// If the table should be created without a rowid (SQLite)
     pub without_rowid: bool,
     /// `LIKE` clause
@@ -4438,7 +4440,9 @@ impl fmt::Display for CreateTable {
         if let Some(sortkey) = &self.sortkey {
             write!(f, " SORTKEY({})", display_comma_separated(sortkey))?;
         }
-        if let Some(query) = &self.query {
+        if let Some(query) = &self.template {
+            write!(f, " USING TEMPLATE ({query})")?;
+        } else if let Some(query) = &self.query {
             write!(f, " AS {query}")?;
         }
         Ok(())

src/ast/helpers/stmt_create_table.rs

@@ -106,6 +106,8 @@ pub struct CreateTableBuilder {
     pub location: Option<String>,
     /// Optional `AS SELECT` query for the table.
     pub query: Option<Box<Query>>,
+    /// Optional Snowflake `USING TEMPLATE` query.
+    pub template: Option<Box<Query>>,
     /// Whether `WITHOUT ROWID` is set.
     pub without_rowid: bool,
     /// Optional `LIKE` clause kind.
@@ -246,6 +248,7 @@ impl CreateTableBuilder {
             file_format: None,
             location: None,
             query: None,
+            template: None,
             without_rowid: false,
             like: None,
             clone: None,
@@ -400,6 +403,11 @@ impl CreateTableBuilder {
         self.query = query;
         self
     }
+    /// Set a Snowflake `USING TEMPLATE` query.
+    pub fn template(mut self, template: Option<Box<Query>>) -> Self {
+        self.template = template;
+        self
+    }
     /// Set `WITHOUT ROWID` option.
     pub fn without_rowid(mut self, without_rowid: bool) -> Self {
         self.without_rowid = without_rowid;
@@ -709,6 +717,7 @@ impl CreateTableBuilder {
             file_format: self.file_format,
             location: self.location,
             query: self.query,
+            template: self.template,
             without_rowid: self.without_rowid,
             like: self.like,
             clone: self.clone,
@@ -808,6 +817,7 @@ impl From<CreateTable> for CreateTableBuilder {
             file_format: table.file_format,
             location: table.location,
             query: table.query,
+            template: table.template,
             without_rowid: table.without_rowid,
             like: table.like,
             clone: table.clone,

src/ast/spans.rs

@@ -716,6 +716,7 @@ impl Spanned for CreateTable {
             snapshot: _,      // bool, BigQuery specific
             name,
             columns,
+            template: _,
             constraints,
             hive_distribution: _, // hive specific
             hive_formats: _,      // hive specific

src/keywords.rs

@@ -1149,6 +1149,7 @@ define_keywords!(
     TASK_AUTO_RETRY_ATTEMPTS,
     TBLPROPERTIES,
     TEMP,
+    TEMPLATE,
     TEMPORARY,
     TEMPTABLE,
     TERMINATED,

src/parser/mod.rs

@@ -9753,8 +9753,17 @@ impl<'a> Parser<'a> {
             None
         };
 
+        let template = if self.parse_keywords(&[Keyword::USING, Keyword::TEMPLATE]) {
+            self.expect_token(&Token::LParen)?;
+            let query = self.parse_query()?;
+            self.expect_token(&Token::RParen)?;
+            Some(query)
+        } else {
+            None
+        };
+
         // Parse optional `AS ( query )`
-        let query = if self.parse_keyword(Keyword::AS) {
+        let query = if template.is_none() && self.parse_keyword(Keyword::AS) {
             Some(self.parse_query()?)
         } else if self.dialect.supports_create_table_select() && self.parse_keyword(Keyword::SELECT)
         {
@@ -9776,6 +9785,7 @@ impl<'a> Parser<'a> {
             .hive_formats(hive_formats)
             .global(global)
             .query(query)
+            .template(template)
             .without_rowid(without_rowid)
             .like(like)
             .clone_clause(clone)

tests/sqlparser_duckdb.rs

@@ -755,6 +755,7 @@ fn test_duckdb_union_datatype() {
             file_format: Default::default(),
             location: Default::default(),
             query: Default::default(),
+            template: Default::default(),
             without_rowid: Default::default(),
             like: Default::default(),
             clone: Default::default(),

tests/sqlparser_mssql.rs

@@ -1982,6 +1982,7 @@ fn parse_create_table_with_valid_options() {
                 file_format: None,
                 location: None,
                 query: None,
+                template: None,
                 without_rowid: false,
                 like: None,
                 clone: None,
@@ -2176,6 +2177,7 @@ fn parse_create_table_with_identity_column() {
                 file_format: None,
                 location: None,
                 query: None,
+                template: None,
                 without_rowid: false,
                 like: None,
                 clone: None,

tests/sqlparser_postgres.rs

@@ -6704,6 +6704,7 @@ fn parse_trigger_related_functions() {
             file_format: None,
             location: None,
             query: None,
+            template: None,
             without_rowid: false,
             like: None,
             clone: None,