Snowflake: DROP STREAM (apache#1973)

Author: yoavcloud

src/ast/mod.rs

@@ -7864,6 +7864,7 @@ pub enum ObjectType {
     Stage,
     Type,
     User,
+    Stream,
 }
 
 impl fmt::Display for ObjectType {
@@ -7880,6 +7881,7 @@ impl fmt::Display for ObjectType {
             ObjectType::Stage => "STAGE",
             ObjectType::Type => "TYPE",
             ObjectType::User => "USER",
+            ObjectType::Stream => "STREAM",
         })
     }
 }

src/keywords.rs

@@ -870,6 +870,7 @@ define_keywords!(
     STORAGE_SERIALIZATION_POLICY,
     STORED,
     STRAIGHT_JOIN,
+    STREAM,
     STRICT,
     STRING,
     STRUCT,

src/parser/mod.rs

@@ -6356,6 +6356,8 @@ impl<'a> Parser<'a> {
             ObjectType::Type
         } else if self.parse_keyword(Keyword::USER) {
             ObjectType::User
+        } else if self.parse_keyword(Keyword::STREAM) {
+            ObjectType::Stream
         } else if self.parse_keyword(Keyword::FUNCTION) {
             return self.parse_drop_function();
         } else if self.parse_keyword(Keyword::POLICY) {

tests/sqlparser_common.rs

@@ -16385,3 +16385,21 @@ fn parse_create_user() {
         _ => unreachable!(),
     }
 }
+
+#[test]
+fn parse_drop_stream() {
+    let sql = "DROP STREAM s1";
+    match verified_stmt(sql) {
+        Statement::Drop {
+            names, object_type, ..
+        } => {
+            assert_eq!(
+                vec!["s1"],
+                names.iter().map(ToString::to_string).collect::<Vec<_>>()
+            );
+            assert_eq!(ObjectType::Stream, object_type);
+        }
+        _ => unreachable!(),
+    }
+    verified_stmt("DROP STREAM IF EXISTS s1");
+}