Merge pull request #174 from matthias-Q/pg_array_function

fix: pg `array` function
DerekStride · Jul 4, 2023 · 85cbb9e · 85cbb9e
2 parents 5beeca2 + 6df5ec8
commit 85cbb9e
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 3 deletions.
diff --git a/grammar.js b/grammar.js
@@ -485,9 +485,18 @@ module.exports = grammar({
 
     array: $ => seq(
       $.keyword_array,
-      "[",
-      comma_list($._expression),
-      "]"
+      choice(
+        seq(
+          "[",
+          comma_list($._expression),
+          "]"
+        ),
+        seq(
+          "(",
+          $._dml_read,
+          ")",
+        )
+      )
     ),
 
     comment: _ => seq('--', /.*/),

diff --git a/test/corpus/select.txt b/test/corpus/select.txt
@@ -2513,6 +2513,34 @@ SELECT array[1, 2, 3 + 4, '5'::int, int_please()];
               (object_reference
                 name: (identifier)))))))))
 
+================================================================================
+Select Array with nested dml_read
+================================================================================
+
+SELECT ARRAY(SELECT col FROM tab)
+
+--------------------------------------------------------------------------------
+
+(program
+  (statement
+    (select
+      (keyword_select)
+      (select_expression
+        (term
+          (array
+            (keyword_array)
+            (select
+              (keyword_select)
+              (select_expression
+                (term
+                  (field
+                    (identifier)))))
+            (from
+              (keyword_from)
+              (relation
+                (object_reference
+                  (identifier))))))))))
+
 ================================================================================
 Select with positive and negative integers
 ================================================================================