Support basic DROP TABLE and DROP VIEW

Author: nickolay

src/sqlast/mod.rs

@@ -347,6 +347,7 @@ pub enum SQLStatement {
         /// VALUES (vector of rows to insert)
         values: Vec<Vec<ASTNode>>,
     },
+    /// COPY (PostgreSQL-specific)
     SQLCopy {
         /// TABLE
         table_name: SQLObjectName,
@@ -371,6 +372,11 @@ pub enum SQLStatement {
         /// WHERE
         selection: Option<ASTNode>,
     },
+    /// DROP
+    SQLDrop {
+        object_type: String,
+        name: SQLObjectName,
+    },
     /// CREATE VIEW
     SQLCreateView {
         /// View name
@@ -468,6 +474,9 @@ impl ToString for SQLStatement {
                 }
                 s
             }
+            SQLStatement::SQLDrop { object_type, name } => {
+                format!("DROP {} {}", object_type, name.to_string())
+            }
             SQLStatement::SQLCreateView {
                 name,
                 query,

src/sqlparser.rs

@@ -99,6 +99,7 @@ impl Parser {
                     "CREATE" => Ok(self.parse_create()?),
                     "DELETE" => Ok(self.parse_delete()?),
                     "INSERT" => Ok(self.parse_insert()?),
+                    "DROP" => Ok(self.parse_drop()?),
                     "ALTER" => Ok(self.parse_alter()?),
                     "COPY" => Ok(self.parse_copy()?),
                     _ => parser_err!(format!(
@@ -838,6 +839,18 @@ impl Parser {
         }
     }
 
+    pub fn parse_drop(&mut self) -> Result<SQLStatement, ParserError> {
+        let object_type = if self.parse_keyword("VIEW") {
+            "VIEW".to_string()
+        } else if self.parse_keyword("TABLE") {
+            "TABLE".to_string()
+        } else {
+            return self.expected("VIEW or TABLE after DROP", self.peek_token());
+        };
+        let name = self.parse_object_name()?;
+        Ok(SQLStatement::SQLDrop { object_type, name })
+    }
+
     pub fn parse_alter(&mut self) -> Result<SQLStatement, ParserError> {
         self.expect_keyword("TABLE")?;
         let _ = self.parse_keyword("ONLY");

tests/sqlparser_common.rs

@@ -620,6 +620,22 @@ fn parse_create_external_table() {
     }
 }
 
+#[test]
+fn parse_drop() {
+    check_one("DROP TABLE mytable", "TABLE", "mytable");
+    check_one("DROP VIEW myview", "VIEW", "myview");
+
+    fn check_one(sql: &str, expected_type: &str, expected_name: &str) {
+        match all_dialects().verified_stmt(sql) {
+            SQLStatement::SQLDrop { name, object_type } => {
+                assert_eq!(name.to_string(), expected_name);
+                assert_eq!(object_type, expected_type);
+            }
+            _ => unreachable!(),
+        }
+    }
+}
+
 #[test]
 fn parse_alter_table_constraint_primary_key() {
     let sql = "ALTER TABLE bazaar.address \