feat: support custom variables ${}

DTStack · Sep 10, 2021 · df18e6c · df18e6c
1 parent dd38dda
commit df18e6c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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+