Merge pull request #2575 from taozhi8833998/fix-insert-into-union-all

taozhi8833998 · web-flow · commit 49e0f4bbd9c8 · 2025-09-19T08:43:52.000+08:00
fix: insert into values clause can be union stmt
diff --git a/pegjs/athena.pegjs b/pegjs/athena.pegjs
@@ -1508,7 +1508,9 @@ set_item
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/bigquery.pegjs b/pegjs/bigquery.pegjs
@@ -586,7 +586,9 @@ insert_partition
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 on_duplicate_update_stmt
   = KW_ON __ 'DUPLICATE'i __ KW_KEY __ KW_UPDATE __ s:set_list {
diff --git a/pegjs/db2.pegjs b/pegjs/db2.pegjs
@@ -1463,7 +1463,9 @@ set_item
   }
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/flinksql.pegjs b/pegjs/flinksql.pegjs
@@ -2198,7 +2198,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/hive.pegjs b/pegjs/hive.pegjs
@@ -1449,7 +1449,9 @@ set_item
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/mariadb.pegjs b/pegjs/mariadb.pegjs
@@ -2631,7 +2631,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:set_op_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/mysql.pegjs b/pegjs/mysql.pegjs
@@ -2892,7 +2892,9 @@ set_item
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:set_op_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/noql.pegjs b/pegjs/noql.pegjs
@@ -3425,7 +3425,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -4124,7 +4124,10 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
+
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/redshift.pegjs b/pegjs/redshift.pegjs
@@ -3453,7 +3453,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/snowflake.pegjs b/pegjs/snowflake.pegjs
@@ -2889,7 +2889,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/sqlite.pegjs b/pegjs/sqlite.pegjs
@@ -1815,7 +1815,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/pegjs/transactsql.pegjs b/pegjs/transactsql.pegjs
@@ -2123,7 +2123,9 @@ set_item
   }
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ il:ident_name_list __ RPAREN {
diff --git a/pegjs/trino.pegjs b/pegjs/trino.pegjs
@@ -2832,7 +2832,9 @@ returning_stmt
 
 insert_value_clause
   = value_clause
-  / select_stmt_nake
+  / u:union_stmt {
+      return u.ast
+  }
 
 insert_partition
   = KW_PARTITION __ LPAREN __ head:ident_name tail:(__ COMMA __ ident_name)* __ RPAREN {
diff --git a/src/insert.js b/src/insert.js
@@ -2,7 +2,7 @@ import { tablesToSQL } from './tables'
 import { exprToSQL } from './expr'
 import { columnRefToSQL } from './column'
 import { identifierToSql, commonOptionConnector, hasVal, toUpper, returningToSQL, literalToSQL } from './util'
-import { selectToSQL } from './select'
+import { unionToSQL } from './union'
 import { setToSQL } from './update'
 
 /**
@@ -11,7 +11,7 @@ import { setToSQL } from './update'
  */
 function valuesToSQL(stmt) {
   const { type } = stmt
-  if (type === 'select') return selectToSQL(stmt)
+  if (type === 'select') return unionToSQL(stmt)
   const values = type === 'values' ? stmt.values : stmt
   const clauses = values.map(value => {
     const sql = exprToSQL(value)
diff --git a/test/postgres.spec.js b/test/postgres.spec.js
@@ -2339,6 +2339,13 @@ describe('Postgres', () => {
           'SELECT CAST(1 AS BIT)',
         ]
       },
+      {
+        title: 'insert into union all stmt',
+        sql: [
+          `truncate table T_LMT_SAMPLE_ALL_DWD_74;insert into T_LMT_SAMPLE_ALL_DWD_74 select 	ID, BLD_STATION_ID, SAMPLE_GENERAL_ID, DON_CODE, DON_FLAG_ID, DON_KIND_ID, SAMPLE_SOURCE_ID, COLLECT_PLACE_ID, SAMPLE_TYPE_ID, SAMPLE_PURPOSE_ID, URGENT_STATUS_ID, BLD_VOLUME, IS_CSEND_SAMPLE, ABO_ID, RHD_ID, REG_DATE, REG_MAN_ID, DEPT_ID, SAMPLE_QUALITY_ID, TEST_POLICY_ID, PROCESS_MODE_ID, AUDIT_ABO_ID, IS_AUDIT_ABO, IS_USED, IS_CREV_SAMPLE, SAMPLE_STATUS_ID, SEND_DATE, COLLECT_DATE, RECEIVE_DATE, NAT_TEST_MODE_ID, NOT_RECEIVE_REASON_ID, CUR_DEPT_ID, SHIPPING_STATUS_ID, LAST_RECEIPT_ID, SEX_ID, IS_CENTRIFUGAL, RECEIVE_MAN_ID, current_date as WAREHOUSE_DEFAULT_TIME from T_LMT_SAMPLE_ODS_81 UNION ALL select ID, BLD_STATION_ID, SAMPLE_GENERAL_ID, DON_CODE, DON_FLAG_ID, DON_KIND_ID, SAMPLE_SOURCE_ID, COLLECT_PLACE_ID, SAMPLE_TYPE_ID, SAMPLE_PURPOSE_ID, URGENT_STATUS_ID, BLD_VOLUME, IS_CSEND_SAMPLE, ABO_ID, RHD_ID, REG_DATE, REG_MAN_ID, DEPT_ID, SAMPLE_QUALITY_ID, TEST_POLICY_ID, PROCESS_MODE_ID, AUDIT_ABO_ID, IS_AUDIT_ABO, IS_USED, IS_CREV_SAMPLE, SAMPLE_STATUS_ID, SEND_DATE, COLLECT_DATE, RECEIVE_DATE, NAT_TEST_MODE_ID, NOT_RECEIVE_REASON_ID, CUR_DEPT_ID, SHIPPING_STATUS_ID, LAST_RECEIPT_ID, SEX_ID, IS_CENTRIFUGAL, RECEIVE_MAN_ID, current_date as WAREHOUSE_DEFAULT_TIME from T_LMT_SAMPLE_STORE_ODS_81`,
+          'TRUNCATE TABLE "T_LMT_SAMPLE_ALL_DWD_74" ; INSERT INTO "T_LMT_SAMPLE_ALL_DWD_74" SELECT ID, BLD_STATION_ID, SAMPLE_GENERAL_ID, DON_CODE, DON_FLAG_ID, DON_KIND_ID, SAMPLE_SOURCE_ID, COLLECT_PLACE_ID, SAMPLE_TYPE_ID, SAMPLE_PURPOSE_ID, URGENT_STATUS_ID, BLD_VOLUME, IS_CSEND_SAMPLE, ABO_ID, RHD_ID, REG_DATE, REG_MAN_ID, DEPT_ID, SAMPLE_QUALITY_ID, TEST_POLICY_ID, PROCESS_MODE_ID, AUDIT_ABO_ID, IS_AUDIT_ABO, IS_USED, IS_CREV_SAMPLE, SAMPLE_STATUS_ID, SEND_DATE, COLLECT_DATE, RECEIVE_DATE, NAT_TEST_MODE_ID, NOT_RECEIVE_REASON_ID, CUR_DEPT_ID, SHIPPING_STATUS_ID, LAST_RECEIPT_ID, SEX_ID, IS_CENTRIFUGAL, RECEIVE_MAN_ID, CURRENT_DATE AS "WAREHOUSE_DEFAULT_TIME" FROM "T_LMT_SAMPLE_ODS_81" UNION ALL SELECT ID, BLD_STATION_ID, SAMPLE_GENERAL_ID, DON_CODE, DON_FLAG_ID, DON_KIND_ID, SAMPLE_SOURCE_ID, COLLECT_PLACE_ID, SAMPLE_TYPE_ID, SAMPLE_PURPOSE_ID, URGENT_STATUS_ID, BLD_VOLUME, IS_CSEND_SAMPLE, ABO_ID, RHD_ID, REG_DATE, REG_MAN_ID, DEPT_ID, SAMPLE_QUALITY_ID, TEST_POLICY_ID, PROCESS_MODE_ID, AUDIT_ABO_ID, IS_AUDIT_ABO, IS_USED, IS_CREV_SAMPLE, SAMPLE_STATUS_ID, SEND_DATE, COLLECT_DATE, RECEIVE_DATE, NAT_TEST_MODE_ID, NOT_RECEIVE_REASON_ID, CUR_DEPT_ID, SHIPPING_STATUS_ID, LAST_RECEIPT_ID, SEX_ID, IS_CENTRIFUGAL, RECEIVE_MAN_ID, CURRENT_DATE AS "WAREHOUSE_DEFAULT_TIME" FROM "T_LMT_SAMPLE_STORE_ODS_81"'
+        ]
+      },
     ]
     neatlyNestTestedSQL(SQL_LIST)
   })
