feat: support DISCARD [ALL | PLANS | SEQUENCES | TEMPORARY | TEMP] (apache#500)

Author: gandronchik

src/ast/mod.rs

@@ -897,6 +897,11 @@ pub enum Statement {
         /// deleted along with the dropped table
         purge: bool,
     },
+    /// DISCARD [ ALL | PLANS | SEQUENCES | TEMPORARY | TEMP ]
+    ///
+    /// Note: this is a PostgreSQL-specific statement,
+    /// but may also compatible with other SQL.
+    Discard { object_type: DiscardObject },
     /// SET [ SESSION | LOCAL ] ROLE role_name
     ///
     /// Note: this is a PostgreSQL-specific statement,
@@ -1561,6 +1566,10 @@ impl fmt::Display for Statement {
                 if *cascade { " CASCADE" } else { "" },
                 if *purge { " PURGE" } else { "" }
             ),
+            Statement::Discard { object_type } => {
+                write!(f, "DISCARD {object_type}", object_type = object_type)?;
+                Ok(())
+            }
             Statement::SetRole {
                 local,
                 session,
@@ -2533,6 +2542,26 @@ impl fmt::Display for MergeClause {
     }
 }
 
+#[derive(Debug, Clone, PartialEq, Eq, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+pub enum DiscardObject {
+    ALL,
+    PLANS,
+    SEQUENCES,
+    TEMP,
+}
+
+impl fmt::Display for DiscardObject {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self {
+            DiscardObject::ALL => f.write_str("ALL"),
+            DiscardObject::PLANS => f.write_str("PLANS"),
+            DiscardObject::SEQUENCES => f.write_str("SEQUENCES"),
+            DiscardObject::TEMP => f.write_str("TEMP"),
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

src/keywords.rs

@@ -184,6 +184,7 @@ define_keywords!(
     DESCRIBE,
     DETERMINISTIC,
     DIRECTORY,
+    DISCARD,
     DISCONNECT,
     DISTINCT,
     DISTRIBUTE,
@@ -371,6 +372,7 @@ define_keywords!(
     PERCENTILE_DISC,
     PERCENT_RANK,
     PERIOD,
+    PLANS,
     PORTION,
     POSITION,
     POSITION_REGEX,

src/parser.rs

@@ -166,6 +166,7 @@ impl<'a> Parser<'a> {
                 Keyword::MSCK => Ok(self.parse_msck()?),
                 Keyword::CREATE => Ok(self.parse_create()?),
                 Keyword::DROP => Ok(self.parse_drop()?),
+                Keyword::DISCARD => Ok(self.parse_discard()?),
                 Keyword::DELETE => Ok(self.parse_delete()?),
                 Keyword::INSERT => Ok(self.parse_insert()?),
                 Keyword::UPDATE => Ok(self.parse_update()?),
@@ -1786,6 +1787,24 @@ impl<'a> Parser<'a> {
         })
     }
 
+    pub fn parse_discard(&mut self) -> Result<Statement, ParserError> {
+        let object_type = if self.parse_keyword(Keyword::ALL) {
+            DiscardObject::ALL
+        } else if self.parse_keyword(Keyword::PLANS) {
+            DiscardObject::PLANS
+        } else if self.parse_keyword(Keyword::SEQUENCES) {
+            DiscardObject::SEQUENCES
+        } else if self.parse_keyword(Keyword::TEMP) || self.parse_keyword(Keyword::TEMPORARY) {
+            DiscardObject::TEMP
+        } else {
+            return self.expected(
+                "ALL, PLANS, SEQUENCES, TEMP or TEMPORARY after DISCARD",
+                self.peek_token(),
+            );
+        };
+        Ok(Statement::Discard { object_type })
+    }
+
     pub fn parse_create_index(&mut self, unique: bool) -> Result<Statement, ParserError> {
         let if_not_exists = self.parse_keywords(&[Keyword::IF, Keyword::NOT, Keyword::EXISTS]);
         let index_name = self.parse_object_name()?;

tests/sqlparser_common.rs

@@ -4787,3 +4787,30 @@ fn parse_is_boolean() {
         res.unwrap_err()
     );
 }
+
+#[test]
+fn parse_discard() {
+    let sql = "DISCARD ALL";
+    match verified_stmt(sql) {
+        Statement::Discard { object_type, .. } => assert_eq!(object_type, DiscardObject::ALL),
+        _ => unreachable!(),
+    }
+
+    let sql = "DISCARD PLANS";
+    match verified_stmt(sql) {
+        Statement::Discard { object_type, .. } => assert_eq!(object_type, DiscardObject::PLANS),
+        _ => unreachable!(),
+    }
+
+    let sql = "DISCARD SEQUENCES";
+    match verified_stmt(sql) {
+        Statement::Discard { object_type, .. } => assert_eq!(object_type, DiscardObject::SEQUENCES),
+        _ => unreachable!(),
+    }
+
+    let sql = "DISCARD TEMP";
+    match verified_stmt(sql) {
+        Statement::Discard { object_type, .. } => assert_eq!(object_type, DiscardObject::TEMP),
+        _ => unreachable!(),
+    }
+}