Support 'timezone_region' for EXTRACT and DATE_PART

apache · Jan 12, 2024 · 8aed07f · 8aed07f
1 parent ce49886
commit 8aed07f
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/src/ast/value.rs b/src/ast/value.rs
@@ -145,6 +145,7 @@ pub enum DateTimeField {
     Timezone,
     TimezoneHour,
     TimezoneMinute,
+    TimezoneRegion,
     NoDateTime,
 }
 
@@ -183,6 +184,7 @@ impl fmt::Display for DateTimeField {
             DateTimeField::Timezone => "TIMEZONE",
             DateTimeField::TimezoneHour => "TIMEZONE_HOUR",
             DateTimeField::TimezoneMinute => "TIMEZONE_MINUTE",
+            DateTimeField::TimezoneRegion => "TIMEZONE_REGION",
             DateTimeField::NoDateTime => "NODATETIME",
         })
     }

diff --git a/src/keywords.rs b/src/keywords.rs
@@ -655,6 +655,7 @@ define_keywords!(
     TIMEZONE,
     TIMEZONE_HOUR,
     TIMEZONE_MINUTE,
+    TIMEZONE_REGION,
     TINYINT,
     TO,
     TOP,

diff --git a/src/parser/mod.rs b/src/parser/mod.rs
@@ -1802,6 +1802,7 @@ impl<'a> Parser<'a> {
                 Keyword::TIMEZONE => Ok(DateTimeField::Timezone),
                 Keyword::TIMEZONE_HOUR => Ok(DateTimeField::TimezoneHour),
                 Keyword::TIMEZONE_MINUTE => Ok(DateTimeField::TimezoneMinute),
+                Keyword::TIMEZONE_REGION => Ok(DateTimeField::TimezoneRegion),
                 _ => self.expected("date/time field", next_token),
             },
             _ => self.expected("date/time field", next_token),

diff --git a/tests/sqlparser_common.rs b/tests/sqlparser_common.rs
@@ -2240,6 +2240,7 @@ fn parse_extract() {
     verified_stmt("SELECT EXTRACT(TIMEZONE FROM d)");
     verified_stmt("SELECT EXTRACT(TIMEZONE_HOUR FROM d)");
     verified_stmt("SELECT EXTRACT(TIMEZONE_MINUTE FROM d)");
+    verified_stmt("SELECT EXTRACT(TIMEZONE_REGION FROM d)");
     verified_stmt("SELECT EXTRACT(TIME FROM d)");
 
     let res = parse_sql_statements("SELECT EXTRACT(JIFFY FROM d)");