Fix deparsing for Array functions (#163)

Example: SELECT ARRAY(SELECT "id" FROM "products")::bigint[]
pganalyze · Dec 27, 2020 · 6989cda · 6989cda
1 parent c8f74aa
commit 6989cda
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/pg_query/deparse.rb b/lib/pg_query/deparse.rb
@@ -1041,6 +1041,8 @@ def deparse_sublink(node)
         format('%s %s ALL (%s)', deparse_item(node['testexpr']), deparse_item(node['operName'][0], :operator), deparse_item(node['subselect']))
       elsif node['subLinkType'] == SUBLINK_TYPE_EXISTS
         format('EXISTS(%s)', deparse_item(node['subselect']))
+      elsif node['subLinkType'] == SUBLINK_TYPE_ARRAY
+        format('ARRAY(%s)', deparse_item(node['subselect']))
       else
         format('(%s)', deparse_item(node['subselect']))
       end

diff --git a/spec/lib/pg_query/deparse_spec.rb b/spec/lib/pg_query/deparse_spec.rb
@@ -175,6 +175,12 @@
         it { is_expected.to eq query }
       end
 
+      context 'ARRAY' do
+        let(:query) { 'SELECT ARRAY(SELECT "id" FROM "products")::bigint[]' }
+
+        it { is_expected.to eq query }
+      end
+
       context 'LATERAL' do
         let(:query) { 'SELECT "m"."name" AS mname, "pname" FROM "manufacturers" m, LATERAL get_product_names("m"."id") pname' }