Merge pull request #43 from sockstack/add-place-holder-support

albin3 · web-flow · commit f2524a0684c4 · 2021-07-30T19:23:57.000+08:00
`${value}` like place holder support
diff --git a/src/sqlParser.jison b/src/sqlParser.jison
@@ -11,6 +11,7 @@
 [#]\s.*\n                                                         /* skip sql comments */
 \s+                                                               /* skip whitespace */
 
+[$][{](.*?)[}]                                                    return 'PLACE_HOLDER'
 [`][a-zA-Z_\u4e00-\u9fa5][a-zA-Z0-9_\u4e00-\u9fa5]*[`]            return 'IDENTIFIER'
 [\w]+[\u4e00-\u9fa5]+[0-9a-zA-Z_\u4e00-\u9fa5]*                   return 'IDENTIFIER'
 [\u4e00-\u9fa5][0-9a-zA-Z_\u4e00-\u9fa5]*                         return 'IDENTIFIER'
@@ -304,6 +305,7 @@ literal
   | number { $$ = $1 }
   | boolean { $$ = $1 }
   | null { $$ = $1 }
+  | place_holder { $$ = $1 }
   ;
 function_call
   : IDENTIFIER '(' function_call_param_list ')' { $$ = {type: 'FunctionCall', name: $1, params: $3} }
@@ -587,3 +589,6 @@ table_factor
   | '(' selectClause ')' aliasOpt { $$ = { type: 'TableFactor', value: { type: 'SubQuery', value: $2 }, alias: $4.alias, hasAs: $4.hasAs} }
   | '(' table_references ')' { $$ = $2; $$.hasParentheses = true }
   ;
+place_holder
+  : PLACE_HOLDER { $$ = { type: 'PlaceHolder', value: $1, param: $1.slice(2, -1)} }
+  ;
diff --git a/src/stringify.js b/src/stringify.js
@@ -551,3 +551,20 @@ Sql.prototype.travelSelectParenthesized = function(ast) {
   this.travel(ast.value);
   this.appendKeyword(')');
 };
+Sql.prototype.travelPlaceHolder = function (ast) {
+  if (ast.left) {
+    this.travel(ast.left);
+  }
+
+  if (ast.operator) {
+    this.append(ast.operator);
+  }
+  
+  if (ast.right) {
+    this.append(ast.right);
+  }
+
+  if (ast.value) {
+    this.travel(ast.value);
+  }
+};
diff --git a/test/main.test.js b/test/main.test.js
@@ -411,6 +411,11 @@ describe('select grammar support', function () {
   it('bugfix table alias2', function () {
     testParser('select a.* from a t1 join b t2 on t1.a = t2.a')
   });
+  it('place holder support', function() {
+    testParser(
+    "select sum(quota_value) value, busi_col2 as sh, ${a} as a, YEAR(now()) from der_quota_summary where table_ename = 'gshmyyszje_derivedidx' and cd = (select id from t1 where a = ${t1})"
+    )
+  });
 
   it('support quoted alias: multiple alias and orderby support', function () {
     testParser('select a as `A A`, b as `B B` from z');
