feat: support custom variables ${}

2021-09-08 20:22:46 +08:00 · 2021-09-08 20:22:46 +08:00 · df18e6cb21
commit df18e6cb21
parent dd38dda19a
1 changed files with 5 additions and 1 deletions
--- a/src/grammar/spark/SparkSql.g4
+++ b/src/grammar/spark/SparkSql.g4
@ -1793,13 +1793,17 @@ BIGDECIMAL_LITERAL
    ;

 IDENTIFIER
-    : (LETTER | DIGIT | '_')+
+    : (LETTER | DIGIT | '_' | CUSTOM_VARS)+
    ;

 BACKQUOTED_IDENTIFIER
    : '`' ( ~'`' | '``' )* '`'
    ;

+CUSTOM_VARS
+    : '${'(IDENTIFIER)'}'
+    ;
+
 fragment DECIMAL_DIGITS
    : DIGIT+ '.' DIGIT*
    | '.' DIGIT+