feat: support arithmetic with in expr

Author: taozhi8833998

pegjs/mariadb.pegjs

@@ -2915,7 +2915,11 @@ not_expr
 comparison_expr
   = left:additive_expr __ rh:comparison_op_right? {
       if (rh === null) return left;
-      else if (rh.type === 'arithmetic') return createBinaryExprChain(left, rh.tail);
+      else if (rh.type === 'arithmetic') {
+        if (!rh.in) return createBinaryExprChain(left, rh.tail);
+        const leftExpr = createBinaryExprChain(left, rh.tail);
+        return createBinaryExpr(rh.in.op, leftExpr, rh.in.right);
+      }
       else return createBinaryExpr(rh.op, left, rh.right);
     }
   / literal_string
@@ -2940,8 +2944,8 @@ comparison_op_right
   / regexp_op_right
 
 arithmetic_op_right
-  = l:(__ arithmetic_comparison_operator __ additive_expr)+ {
-      return { type: 'arithmetic', tail: l };
+  = l:(__ arithmetic_comparison_operator __ additive_expr)+ __ i:in_op_right? {
+      return { type: 'arithmetic', tail: l, in: i };
     }
 
 arithmetic_comparison_operator

pegjs/mysql.pegjs

@@ -3177,7 +3177,11 @@ not_expr
 comparison_expr
   = left:additive_expr __ rh:comparison_op_right? {
       if (rh === null) return left;
-      else if (rh.type === 'arithmetic') return createBinaryExprChain(left, rh.tail);
+      else if (rh.type === 'arithmetic') {
+        if (!rh.in) return createBinaryExprChain(left, rh.tail);
+        const leftExpr = createBinaryExprChain(left, rh.tail);
+        return createBinaryExpr(rh.in.op, leftExpr, rh.in.right);
+      }
       else return createBinaryExpr(rh.op, left, rh.right);
     }
   / literal_string
@@ -3202,8 +3206,8 @@ comparison_op_right
   / regexp_op_right
 
 arithmetic_op_right
-  = l:(__ arithmetic_comparison_operator __ additive_expr)+ {
-      return { type: 'arithmetic', tail: l };
+  = l:(__ arithmetic_comparison_operator __ additive_expr)+ __ i:in_op_right? {
+      return { type: 'arithmetic', tail: l, in: i };
     }
 
 arithmetic_comparison_operator

test/mysql-mariadb.spec.js

@@ -1284,6 +1284,13 @@ describe('mysql', () => {
           'SELECT * FROM `A` INNER JOIN (`B` INNER JOIN `C` ON `B`.`x` = `C`.`x`) ON `A`.`y` = `C`.`y`'
         ]
       },
+      {
+        title: 'arithmetic and in expr',
+        sql: [
+          'SELECT * FROM T1 WHERE a = b IN (SELECT flag FROM T2)',
+          'SELECT * FROM `T1` WHERE `a` = `b` IN (SELECT `flag` FROM `T2`)'
+        ]
+      },
     ]
     SQL_LIST.forEach(sqlInfo => {
       const { title, sql } = sqlInfo