fix: update set item support full expr in all db

taozhi8833998 · taozhi8833998 · commit f28e60d1f46c · 2025-07-01T09:08:24.000+08:00
diff --git a/pegjs/athena.pegjs b/pegjs/athena.pegjs
@@ -1496,7 +1496,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
     }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/bigquery.pegjs b/pegjs/bigquery.pegjs
@@ -552,7 +552,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
   }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/db2.pegjs b/pegjs/db2.pegjs
@@ -1452,7 +1452,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
     }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/flinksql.pegjs b/pegjs/flinksql.pegjs
@@ -2177,7 +2177,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       // => { column: ident; value: additive_expr; table?: ident;}
       return { column: c, value: v, table: tbl && tbl[0] };
     }
diff --git a/pegjs/hive.pegjs b/pegjs/hive.pegjs
@@ -1437,7 +1437,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
     }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/mariadb.pegjs b/pegjs/mariadb.pegjs
@@ -2609,7 +2609,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
     }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/mysql.pegjs b/pegjs/mysql.pegjs
@@ -2878,7 +2878,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
   }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/noql.pegjs b/pegjs/noql.pegjs
@@ -3404,7 +3404,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       // => { column: ident; value: additive_expr; table?: ident;}
       return { column: c, value: v, table: tbl && tbl[0] };
     }
diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -4095,7 +4095,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = c:column_ref_array_index __ '=' __ v:additive_expr {
+  = c:column_ref_array_index __ '=' __ v:expr {
     // => { column: ident; value: additive_expr; table?: ident;}
     return {  ...c, value: v };
   }
diff --git a/pegjs/redshift.pegjs b/pegjs/redshift.pegjs
@@ -3432,7 +3432,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw_type __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw_type __ '=' __ v:expr {
       // => { column: ident; value: additive_expr; table?: ident;}
       return { column: { expr: c }, value: v, table: tbl && tbl[0] };
     }
diff --git a/pegjs/snowflake.pegjs b/pegjs/snowflake.pegjs
@@ -2868,7 +2868,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       // => { column: ident; value: additive_expr; table?: ident;}
       return { column: c, value: v, table: tbl && tbl[0] };
     }
diff --git a/pegjs/sqlite.pegjs b/pegjs/sqlite.pegjs
@@ -1793,7 +1793,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
   }
   / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/transactsql.pegjs b/pegjs/transactsql.pegjs
@@ -2113,7 +2113,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {
       return { column: c, value: v, table: tbl && tbl[0] };
     }
     / tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {
diff --git a/pegjs/trino.pegjs b/pegjs/trino.pegjs
@@ -2810,7 +2810,7 @@ set_list
  * 'col1 = (col2 > 3)'
  */
 set_item
-  = tbl:(ident __ DOT)? __ c:column_without_kw_type __ '=' __ v:additive_expr {
+  = tbl:(ident __ DOT)? __ c:column_without_kw_type __ '=' __ v:expr {
       // => { column: ident; value: additive_expr; table?: ident;}
       return { column: { expr: c }, value: v, table: tbl && tbl[0] };
     }
diff --git a/test/postgres.spec.js b/test/postgres.spec.js
@@ -1875,6 +1875,13 @@ describe('Postgres', () => {
         `SELECT * FROM "users" WHERE id = E'one'`
       ]
     },
+    {
+      title: 'set item support full expr',
+      sql: [
+        'update api.entities set is_active = not is_active WHERE id = $1',
+        'UPDATE "api"."entities" SET is_active = NOT is_active WHERE id = $1'
+      ]
+    },
   ]
   function neatlyNestTestedSQL(sqlList){
     sqlList.forEach(sqlInfo => {

Original file line number	Diff line number	Diff line change
`@@ -1496,7 +1496,7 @@ set_list`
`1496`	`1496`	`* 'col1 = (col2 > 3)'`
`1497`	`1497`	`*/`
`1498`	`1498`	`set_item`
`1499`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`1499`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`1500`	`1500`	`return { column: c, value: v, table: tbl && tbl[0] };`
`1501`	`1501`	`}`
`1502`	`1502`	`/ tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {`
Original file line number	Diff line number	Diff line change
`@@ -552,7 +552,7 @@ set_list`
`552`	`552`	`* 'col1 = (col2 > 3)'`
`553`	`553`	`*/`
`554`	`554`	`set_item`
`555`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`555`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`556`	`556`	`return { column: c, value: v, table: tbl && tbl[0] };`
`557`	`557`	`}`
`558`	`558`	`/ tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {`
Original file line number	Diff line number	Diff line change
`@@ -1452,7 +1452,7 @@ set_list`
`1452`	`1452`	`* 'col1 = (col2 > 3)'`
`1453`	`1453`	`*/`
`1454`	`1454`	`set_item`
`1455`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`1455`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`1456`	`1456`	`return { column: c, value: v, table: tbl && tbl[0] };`
`1457`	`1457`	`}`
`1458`	`1458`	`/ tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {`
Original file line number	Diff line number	Diff line change
`@@ -2177,7 +2177,7 @@ set_list`
`2177`	`2177`	`* 'col1 = (col2 > 3)'`
`2178`	`2178`	`*/`
`2179`	`2179`	`set_item`
`2180`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`2180`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`2181`	`2181`	`// => { column: ident; value: additive_expr; table?: ident;}`
`2182`	`2182`	`return { column: c, value: v, table: tbl && tbl[0] };`
`2183`	`2183`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1437,7 +1437,7 @@ set_list`
`1437`	`1437`	`* 'col1 = (col2 > 3)'`
`1438`	`1438`	`*/`
`1439`	`1439`	`set_item`
`1440`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`1440`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`1441`	`1441`	`return { column: c, value: v, table: tbl && tbl[0] };`
`1442`	`1442`	`}`
`1443`	`1443`	`/ tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {`
Original file line number	Diff line number	Diff line change
`@@ -2609,7 +2609,7 @@ set_list`
`2609`	`2609`	`* 'col1 = (col2 > 3)'`
`2610`	`2610`	`*/`
`2611`	`2611`	`set_item`
`2612`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`2612`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`2613`	`2613`	`return { column: c, value: v, table: tbl && tbl[0] };`
`2614`	`2614`	`}`
`2615`	`2615`	`/ tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {`
Original file line number	Diff line number	Diff line change
`@@ -2878,7 +2878,7 @@ set_list`
`2878`	`2878`	`* 'col1 = (col2 > 3)'`
`2879`	`2879`	`*/`
`2880`	`2880`	`set_item`
`2881`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`2881`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`2882`	`2882`	`return { column: c, value: v, table: tbl && tbl[0] };`
`2883`	`2883`	`}`
`2884`	`2884`	`/ tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ KW_VALUES __ LPAREN __ v:column_ref __ RPAREN {`
Original file line number	Diff line number	Diff line change
`@@ -3404,7 +3404,7 @@ set_list`
`3404`	`3404`	`* 'col1 = (col2 > 3)'`
`3405`	`3405`	`*/`
`3406`	`3406`	`set_item`
`3407`		`- = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:additive_expr {`
	`3407`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw __ '=' __ v:expr {`
`3408`	`3408`	`// => { column: ident; value: additive_expr; table?: ident;}`
`3409`	`3409`	`return { column: c, value: v, table: tbl && tbl[0] };`
`3410`	`3410`	`}`
Original file line number	Diff line number	Diff line change
`@@ -4095,7 +4095,7 @@ set_list`
`4095`	`4095`	`* 'col1 = (col2 > 3)'`
`4096`	`4096`	`*/`
`4097`	`4097`	`set_item`
`4098`		`- = c:column_ref_array_index __ '=' __ v:additive_expr {`
	`4098`	`+ = c:column_ref_array_index __ '=' __ v:expr {`
`4099`	`4099`	`// => { column: ident; value: additive_expr; table?: ident;}`
`4100`	`4100`	`return { ...c, value: v };`
`4101`	`4101`	`}`
Original file line number	Diff line number	Diff line change
`@@ -3432,7 +3432,7 @@ set_list`
`3432`	`3432`	`* 'col1 = (col2 > 3)'`
`3433`	`3433`	`*/`
`3434`	`3434`	`set_item`
`3435`		`- = tbl:(ident __ DOT)? __ c:column_without_kw_type __ '=' __ v:additive_expr {`
	`3435`	`+ = tbl:(ident __ DOT)? __ c:column_without_kw_type __ '=' __ v:expr {`
`3436`	`3436`	`// => { column: ident; value: additive_expr; table?: ident;}`
`3437`	`3437`	`return { column: { expr: c }, value: v, table: tbl && tbl[0] };`
`3438`	`3438`	`}`