Add support for + char in Snowflake stage names

Author: yoavcloud

src/dialect/snowflake.rs

@@ -718,6 +718,7 @@ pub fn parse_stage_name_identifier(parser: &mut Parser) -> Result<Ident, ParserE
             Token::Tilde => ident.push('~'),
             Token::Mod => ident.push('%'),
             Token::Div => ident.push('/'),
+            Token::Plus => ident.push('+'),
             Token::Word(w) => ident.push_str(&w.to_string()),
             _ => return parser.expected("stage name identifier", parser.peek_token()),
         }

tests/sqlparser_snowflake.rs

@@ -2581,6 +2581,26 @@ fn test_snowflake_copy_into() {
         }
         _ => unreachable!(),
     }
+
+    // Test for non-ident characters in stage names
+    let sql = "COPY INTO a.b FROM @namespace.stage_name/x@x~x%x+";
+    assert_eq!(snowflake().verified_stmt(sql).to_string(), sql);
+    match snowflake().verified_stmt(sql) {
+        Statement::CopyIntoSnowflake { into, from_obj, .. } => {
+            assert_eq!(
+                into,
+                ObjectName::from(vec![Ident::new("a"), Ident::new("b")])
+            );
+            assert_eq!(
+                from_obj,
+                Some(ObjectName::from(vec![
+                    Ident::new("@namespace"),
+                    Ident::new("stage_name/x@x~x%x+")
+                ]))
+            )
+        }
+        _ => unreachable!(),
+    }
 }
 
 #[test]
@@ -4231,4 +4251,4 @@ fn test_snowflake_create_view_with_composite_policy_name() {
     let create_view_with_tag =
         r#"CREATE VIEW X (COL WITH MASKING POLICY foo.bar.baz) AS SELECT * FROM Y"#;
     snowflake().verified_stmt(create_view_with_tag);
-}
+}