feat: list stage name to plain_identifier.

databendlabs · Sep 25, 2024 · 4b97516 · 4b97516
1 parent a402d6c
commit 4b97516
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/query/ast/src/parser/common.rs b/src/query/ast/src/parser/common.rs
@@ -102,7 +102,7 @@ pub fn stage_name(i: Input) -> IResult<Identifier> {
     });
 
     rule!(
-        #ident
+        #plain_identifier(|token| token.is_reserved_ident(false))(i)
         | #anonymous_stage
     )(i)
 }

diff --git a/src/query/ast/src/parser/token.rs b/src/query/ast/src/parser/token.rs
@@ -157,7 +157,10 @@ pub enum TokenKind {
     #[regex(r#"\$\$([^\$]|(\$[^\$]))*\$\$"#)]
     LiteralCodeString,
 
-    #[regex(r#"@([^\s`;'"()]|\\\s|\\'|\\"|\\\\)+"#)]
+    #[regex(r#"@[_a-zA-Z][_$a-zA-Z0-9]*"#)]
+    #[regex(r#"@`[^`]*`"#)]
+    #[regex(r#"@"([^"\\]|\\.|"")*""#)]
+    #[regex(r#"@'([^'\\]|\\.|'')*'"#)]
     LiteralAtString,
 
     #[regex(r"[xX]'[a-fA-F0-9]*'")]