snowflake: qualify

Author: donhcd

src/ast/query.rs

@@ -132,6 +132,8 @@ pub struct Select {
     pub group_by: Vec<Expr>,
     /// HAVING
     pub having: Option<Expr>,
+    /// QUALIFY https://docs.snowflake.com/en/sql-reference/constructs/qualify.html
+    pub qualify: Option<Expr>,
 }
 
 impl fmt::Display for Select {
@@ -153,6 +155,9 @@ impl fmt::Display for Select {
         if let Some(ref having) = self.having {
             write!(f, " HAVING {}", having)?;
         }
+        if let Some(ref qualify) = self.qualify {
+            write!(f, " QUALIFY {}", qualify)?;
+        }
         Ok(())
     }
 }

src/dialect/keywords.rs

@@ -332,6 +332,7 @@ define_keywords!(
     PREPARE,
     PRIMARY,
     PROCEDURE,
+    QUALIFY,
     QUARTER,
     RANGE,
     RANK,

src/parser.rs

@@ -2057,6 +2057,12 @@ impl<'a> Parser<'a> {
             None
         };
 
+        let qualify = if self.parse_keyword(Keyword::QUALIFY) {
+            Some(self.parse_expr()?)
+        } else {
+            None
+        };
+
         Ok(Select {
             distinct,
             top,
@@ -2065,6 +2071,7 @@ impl<'a> Parser<'a> {
             selection,
             group_by,
             having,
+            qualify,
         })
     }