Refactored into struct DropDomain

Author: LucaCappelletti94

src/ast/mod.rs

@@ -3325,11 +3325,7 @@ pub enum Statement {
     ///
     /// DROP DOMAIN [ IF EXISTS ] name [, ...] [ CASCADE | RESTRICT ]
     ///
-    DropDomain {
-        if_exists: bool,
-        name: ObjectName,
-        drop_behavior: Option<DropBehavior>,
-    },
+    DropDomain(DropDomain),
     /// ```sql
     /// DROP PROCEDURE
     /// ```
@@ -5104,11 +5100,11 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
-            Statement::DropDomain {
+            Statement::DropDomain(DropDomain {
                 if_exists,
                 name,
                 drop_behavior,
-            } => {
+            }) => {
                 write!(
                     f,
                     "DROP DOMAIN{} {name}",
@@ -6854,6 +6850,19 @@ impl fmt::Display for CloseCursor {
     }
 }
 
+/// A Drop Domain statement
+#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
+pub struct DropDomain {
+    /// Whether to drop the domain if it exists
+    pub if_exists: bool,
+    /// The name of the domain to drop
+    pub name: ObjectName,
+    /// The behavior to apply when dropping the domain
+    pub drop_behavior: Option<DropBehavior>,
+}
+
 /// A function call
 #[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
 #[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]

src/parser/mod.rs

@@ -6155,11 +6155,11 @@ impl<'a> Parser<'a> {
         let if_exists = self.parse_keywords(&[Keyword::IF, Keyword::EXISTS]);
         let name = self.parse_object_name(false)?;
         let drop_behavior = self.parse_optional_drop_behavior();
-        Ok(Statement::DropDomain {
+        Ok(Statement::DropDomain(DropDomain {
             if_exists,
             name,
             drop_behavior,
-        })
+        }))
     }
 
     /// ```sql

tests/sqlparser_postgres.rs

@@ -4244,58 +4244,58 @@ fn parse_drop_domain() {
     let sql = "DROP DOMAIN IF EXISTS jpeg_domain";
     assert_eq!(
         pg().verified_stmt(sql),
-        Statement::DropDomain {
+        Statement::DropDomain(DropDomain {
             if_exists: true,
             name: ObjectName::from(vec![Ident {
                 value: "jpeg_domain".to_string(),
                 quote_style: None,
                 span: Span::empty(),
             }]),
             drop_behavior: None
-        }
+        })
     );
 
     let sql = "DROP DOMAIN jpeg_domain";
     assert_eq!(
         pg().verified_stmt(sql),
-        Statement::DropDomain {
+        Statement::DropDomain(DropDomain {
             if_exists: false,
             name: ObjectName::from(vec![Ident {
                 value: "jpeg_domain".to_string(),
                 quote_style: None,
                 span: Span::empty(),
             }]),
             drop_behavior: None
-        }
+        })
     );
 
     let sql = "DROP DOMAIN IF EXISTS jpeg_domain CASCADE";
     assert_eq!(
         pg().verified_stmt(sql),
-        Statement::DropDomain {
+        Statement::DropDomain(DropDomain {
             if_exists: true,
             name: ObjectName::from(vec![Ident {
                 value: "jpeg_domain".to_string(),
                 quote_style: None,
                 span: Span::empty(),
             }]),
             drop_behavior: Some(DropBehavior::Cascade)
-        }
+        })
     );
 
     let sql = "DROP DOMAIN IF EXISTS jpeg_domain RESTRICT";
 
     assert_eq!(
         pg().verified_stmt(sql),
-        Statement::DropDomain {
+        Statement::DropDomain(DropDomain {
             if_exists: true,
             name: ObjectName::from(vec![Ident {
                 value: "jpeg_domain".to_string(),
                 quote_style: None,
                 span: Span::empty(),
             }]),
             drop_behavior: Some(DropBehavior::Restrict)
-        }
+        })
     );
 }