Feat(bigquery): add support for CREATE TABLE .. COPY DDL syntax (#2305)

tobymao · Sep 23, 2023 · 8ed0a81 · 8ed0a81
1 parent e1cbff4
commit 8ed0a81
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/sqlglot/dialects/bigquery.py b/sqlglot/dialects/bigquery.py
@@ -419,6 +419,7 @@ class Generator(generator.Generator):
         RENAME_TABLE_WITH_DB = False
         NVL2_SUPPORTED = False
         UNNEST_WITH_ORDINALITY = False
+        CLONE_KEYWORD = "COPY"
 
         TRANSFORMS = {
             **generator.Generator.TRANSFORMS,

diff --git a/sqlglot/generator.py b/sqlglot/generator.py
@@ -204,6 +204,9 @@ class Generator:
     # Whether or not session variables / parameters are supported, e.g. @x in T-SQL
     SUPPORTS_PARAMETERS = True
 
+    # The keyword used to clone a table in a DDL statement
+    CLONE_KEYWORD = "CLONE"
+
     TYPE_MAPPING = {
         exp.DataType.Type.NCHAR: "CHAR",
         exp.DataType.Type.NVARCHAR: "VARCHAR",
@@ -817,7 +820,7 @@ def create_sql(self, expression: exp.Create) -> str:
     def clone_sql(self, expression: exp.Clone) -> str:
         this = self.sql(expression, "this")
         shallow = "SHALLOW " if expression.args.get("shallow") else ""
-        this = f"{shallow}CLONE {this}"
+        this = f"{shallow}{self.CLONE_KEYWORD} {this}"
         when = self.sql(expression, "when")
 
         if when:

diff --git a/sqlglot/parser.py b/sqlglot/parser.py
@@ -847,6 +847,7 @@ class Parser(metaclass=_Parser):
 
     INSERT_ALTERNATIVES = {"ABORT", "FAIL", "IGNORE", "REPLACE", "ROLLBACK"}
 
+    CLONE_KEYWORDS = {"CLONE", "COPY"}
     CLONE_KINDS = {"TIMESTAMP", "OFFSET", "STATEMENT"}
 
     TABLE_INDEX_HINT_TOKENS = {TokenType.FORCE, TokenType.IGNORE, TokenType.USE}
@@ -1344,7 +1345,7 @@ def extend_props(temp_props: t.Optional[exp.Properties]) -> None:
 
             shallow = self._match_text_seq("SHALLOW")
 
-            if self._match_text_seq("CLONE"):
+            if self._match_texts(self.CLONE_KEYWORDS):
                 clone = self._parse_table(schema=True)
                 when = self._match_texts({"AT", "BEFORE"}) and self._prev.text.upper()
                 clone_kind = (

diff --git a/tests/dialects/test_bigquery.py b/tests/dialects/test_bigquery.py
@@ -122,6 +122,10 @@ def test_bigquery(self):
             """SELECT JSON '"foo"' AS json_data""",
             """SELECT PARSE_JSON('"foo"') AS json_data""",
         )
+        self.validate_identity(
+            "CREATE OR REPLACE TABLE `a.b.c` copy `a.b.d`",
+            "CREATE OR REPLACE TABLE a.b.c COPY a.b.d",
+        )
 
         self.validate_all("SELECT SPLIT(foo)", write={"bigquery": "SELECT SPLIT(foo, ',')"})
         self.validate_all("SELECT 1 AS hash", write={"bigquery": "SELECT 1 AS `hash`"})