From 784605c9132d7e14f9e3b396c26d63af086ff885 Mon Sep 17 00:00:00 2001
From: Yoav Cohen <59807311+yoavcloud@users.noreply.github.com>
Date: Wed, 29 Jan 2025 17:50:36 +0100
Subject: [PATCH] Fix bug when parsing a Snowflake stage with `;` suffix
 (#1688)

---
 src/dialect/snowflake.rs     | 2 +-
 tests/sqlparser_snowflake.rs | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/dialect/snowflake.rs b/src/dialect/snowflake.rs
index eb8ea4de..d775ffc3 100644
--- a/src/dialect/snowflake.rs
+++ b/src/dialect/snowflake.rs
@@ -625,7 +625,7 @@ pub fn parse_stage_name_identifier(parser: &mut Parser) -> Result<Ident, ParserE
     let mut ident = String::new();
     while let Some(next_token) = parser.next_token_no_skip() {
         match &next_token.token {
-            Token::Whitespace(_) => break,
+            Token::Whitespace(_) | Token::SemiColon => break,
             Token::Period => {
                 parser.prev_token();
                 break;
diff --git a/tests/sqlparser_snowflake.rs b/tests/sqlparser_snowflake.rs
index f8b58dd1..1310b984 100644
--- a/tests/sqlparser_snowflake.rs
+++ b/tests/sqlparser_snowflake.rs
@@ -3102,6 +3102,10 @@ fn parse_ls_and_rm() {
     };
 
     snowflake().verified_stmt(r#"LIST @"STAGE_WITH_QUOTES""#);
+    // Semi-colon after stage name - should terminate the stage name
+    snowflake()
+        .parse_sql_statements("LIST @db1.schema1.stage1/dir1/;")
+        .unwrap();
 }
 
 #[test]