bigquery: PRIMARY KEY NOT ENFORCED

Author: lustefaniak

src/ast/ddl.rs

@@ -806,6 +806,7 @@ pub enum ConstraintProperty {
     NoValidate,
     Rely,
     NoRely,
+    NotEnforced,
 }
 
 impl fmt::Display for ConstraintProperty {
@@ -815,6 +816,7 @@ impl fmt::Display for ConstraintProperty {
             ConstraintProperty::NoValidate => "NOVALIDATE",
             ConstraintProperty::Rely => "RELY",
             ConstraintProperty::NoRely => "NORELY",
+            ConstraintProperty::NotEnforced => "NOT ENFORCED",
         })
     }
 }

src/keywords.rs

@@ -243,6 +243,7 @@ define_keywords!(
     ENDPOINT,
     END_FRAME,
     END_PARTITION,
+    ENFORCED,
     ENGINE,
     ENUM,
     EPOCH,

src/parser/mod.rs

@@ -4487,6 +4487,15 @@ impl<'a> Parser<'a> {
                 }
             }
         }
+        if dialect_of!(self is BigQueryDialect) {
+            loop {
+                if self.parse_keywords(&[Keyword::NOT, Keyword::ENFORCED]) {
+                    constraint_properties.push(ConstraintProperty::NotEnforced)
+                } else {
+                    break;
+                }
+            }
+        }
         Ok(constraint_properties)
     }
 

tests/sqlparser_bigquery.rs

@@ -1371,3 +1371,10 @@ fn test_create_table_cluster_by() {
         "CREATE TABLE `myproject`.`mydataset`.`mytable` (service_id STRING, account_id STRING, state STRING, valid_from TIMESTAMP, valid_to TIMESTAMP) PARTITION BY TIMESTAMP_TRUNC(valid_from, MONTH) CLUSTER BY (account_id, state) OPTIONS (description = \"State of an service at a point in time\")"
     );
 }
+
+#[test]
+fn test_create_table_primary_key_not_enforced() {
+    bigquery().verified_stmt(
+        "CREATE TABLE `myproject`.`mydataset`.`mytable` (id INT64, PRIMARY KEY (id) NOT ENFORCED)",
+    );
+}