feat: Add set operation (select ... union select ...) for insert stat…

…ements.
DerekStride · Oct 8, 2023 · 970b548 · 970b548
1 parent 385aff4
commit 970b548
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/grammar.js b/grammar.js
@@ -1907,7 +1907,7 @@ module.exports = grammar({
           $.keyword_values,
           comma_list($.list, true),
         ),
-        $._select_statement,
+        $._dml_read,
       ),
     ),
 

diff --git a/test/corpus/insert.txt b/test/corpus/insert.txt
@@ -526,3 +526,38 @@ FROM
                     (identifier))
                   (literal)))))
           (identifier))))))
+
+================================================================================
+Insert from unioned select
+================================================================================
+
+INSERT INTO some_table
+  (field)
+  (SELECT "String value"
+   UNION
+   SELECT "String value");
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (insert
+      (keyword_insert)
+      (keyword_into)
+      (object_reference
+        (identifier))
+      (list
+        (column
+          (identifier)))
+      (set_operation
+        (select
+          (keyword_select)
+          (select_expression
+            (term
+              (literal))))
+        (keyword_union)
+        (select
+          (keyword_select)
+          (select_expression
+            (term
+              (literal))))))))