bigquery: EXTRACT support For DAYOFWEEK, DAYOFYEAR, ISOWEEK, TIME

Author: lustefaniak

src/ast/value.rs

@@ -117,6 +117,8 @@ pub enum DateTimeField {
     Month,
     Week,
     Day,
+    DayOfWeek,
+    DayOfYear,
     Date,
     Hour,
     Minute,
@@ -127,6 +129,7 @@ pub enum DateTimeField {
     Doy,
     Epoch,
     Isodow,
+    IsoWeek,
     Isoyear,
     Julian,
     Microsecond,
@@ -138,6 +141,7 @@ pub enum DateTimeField {
     Nanosecond,
     Nanoseconds,
     Quarter,
+    Time,
     Timezone,
     TimezoneHour,
     TimezoneMinute,
@@ -151,6 +155,8 @@ impl fmt::Display for DateTimeField {
             DateTimeField::Month => "MONTH",
             DateTimeField::Week => "WEEK",
             DateTimeField::Day => "DAY",
+            DateTimeField::DayOfWeek => "DAYOFWEEK",
+            DateTimeField::DayOfYear => "DAYOFYEAR",
             DateTimeField::Date => "DATE",
             DateTimeField::Hour => "HOUR",
             DateTimeField::Minute => "MINUTE",
@@ -162,6 +168,7 @@ impl fmt::Display for DateTimeField {
             DateTimeField::Epoch => "EPOCH",
             DateTimeField::Isodow => "ISODOW",
             DateTimeField::Isoyear => "ISOYEAR",
+            DateTimeField::IsoWeek => "ISOWEEK",
             DateTimeField::Julian => "JULIAN",
             DateTimeField::Microsecond => "MICROSECOND",
             DateTimeField::Microseconds => "MICROSECONDS",
@@ -172,6 +179,7 @@ impl fmt::Display for DateTimeField {
             DateTimeField::Nanosecond => "NANOSECOND",
             DateTimeField::Nanoseconds => "NANOSECONDS",
             DateTimeField::Quarter => "QUARTER",
+            DateTimeField::Time => "TIME",
             DateTimeField::Timezone => "TIMEZONE",
             DateTimeField::TimezoneHour => "TIMEZONE_HOUR",
             DateTimeField::TimezoneMinute => "TIMEZONE_MINUTE",

src/keywords.rs

@@ -196,6 +196,8 @@ define_keywords!(
     DATE,
     DATETIME,
     DAY,
+    DAYOFWEEK,
+    DAYOFYEAR,
     DEALLOCATE,
     DEC,
     DECADE,
@@ -334,6 +336,7 @@ define_keywords!(
     IS,
     ISODOW,
     ISOLATION,
+    ISOWEEK,
     ISOYEAR,
     JAR,
     JOIN,

src/parser/mod.rs

@@ -1503,6 +1503,8 @@ impl<'a> Parser<'a> {
                 Keyword::MONTH => Ok(DateTimeField::Month),
                 Keyword::WEEK => Ok(DateTimeField::Week),
                 Keyword::DAY => Ok(DateTimeField::Day),
+                Keyword::DAYOFWEEK => Ok(DateTimeField::DayOfWeek),
+                Keyword::DAYOFYEAR => Ok(DateTimeField::DayOfYear),
                 Keyword::DATE => Ok(DateTimeField::Date),
                 Keyword::HOUR => Ok(DateTimeField::Hour),
                 Keyword::MINUTE => Ok(DateTimeField::Minute),
@@ -1514,6 +1516,7 @@ impl<'a> Parser<'a> {
                 Keyword::EPOCH => Ok(DateTimeField::Epoch),
                 Keyword::ISODOW => Ok(DateTimeField::Isodow),
                 Keyword::ISOYEAR => Ok(DateTimeField::Isoyear),
+                Keyword::ISOWEEK => Ok(DateTimeField::IsoWeek),
                 Keyword::JULIAN => Ok(DateTimeField::Julian),
                 Keyword::MICROSECOND => Ok(DateTimeField::Microsecond),
                 Keyword::MICROSECONDS => Ok(DateTimeField::Microseconds),
@@ -1524,6 +1527,7 @@ impl<'a> Parser<'a> {
                 Keyword::NANOSECOND => Ok(DateTimeField::Nanosecond),
                 Keyword::NANOSECONDS => Ok(DateTimeField::Nanoseconds),
                 Keyword::QUARTER => Ok(DateTimeField::Quarter),
+                Keyword::TIME => Ok(DateTimeField::Time),
                 Keyword::TIMEZONE => Ok(DateTimeField::Timezone),
                 Keyword::TIMEZONE_HOUR => Ok(DateTimeField::TimezoneHour),
                 Keyword::TIMEZONE_MINUTE => Ok(DateTimeField::TimezoneMinute),