pg_conn 0.26.0 → 0.26.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4148bbd8c21613249beed07d5248aa71780a8805881afc94899f9f29cd2d452d
-  data.tar.gz: 21bea1c8c108a4a135eb1d6936a56995b457db0e8192a17b745d0b3dd1676846
+  metadata.gz: 6dc75936f3c9bd5db772a2f8a80aa8e8bb25cbd1ae2181492766a7ed6ab9104a
+  data.tar.gz: 8ac094777a20c074f4a018b4bb0ea53efff219eb2401e18bf9c5887721fbb1fe
 SHA512:
-  metadata.gz: 7cca5aea976d949607b1650e8f00347a22be14907c8296a26379cd804a51f5610bb9786bc21b3b9317496a00ef135564033f276d5e3300dbd4d133b056835a79
-  data.tar.gz: 8bcf303b9712c1e6de69c301115c88a53d908aad90c8bce6b8e73e912d6b710772fb88a6bb191bc34142ca29f016845bfbc035cda4670c91d5e3d407ce82907e
+  metadata.gz: 8651be5de05b08d991cc257dab9cc060c4dfe22f8c22b2489f7e9ecc3654d51c55ad8494287b1cb829dc92a50cf60eb777b8814bcc23b348143caf9339729a8c
+  data.tar.gz: 8c11d1e8cb3ceb859438e9ccb98fe38de9ed9126c449c58972cef97d9fb3108c87ea2a18899aea9d500eaa249625d0a1a783176e3fd4e8bc7ebe035cec31f67b

data/lib/pg_conn/version.rb

@@ -1,3 +1,3 @@
 module PgConn
-  VERSION = "0.26.0"
+  VERSION = "0.26.1"
 end

data/lib/pg_conn.rb

@@ -315,41 +315,21 @@ module PgConn
     end
 
     # Quote a record and cast it into the given type, the type can also be a
-    # table or view. 'data' is a hash or struct representation of the record
+    # table or view. 'data' is an array, hash, or struct representation of the
+    # record
     #
     # Note that the fields are retrived from the database so this method is not
     # as fast as the other quote-methods. It is however very convenient when
     # you're testing and need a composite type because record-quoting can
     # easily become unwieldly
-    def quote_record(schema_name = nil, type, data, elem_types: nil)
-      qual_name = [schema_name, type].compact .join('.')
-
-      fields = self.values(%(
-        select attname
-        from pg_attribute
-        where
-          attrelid = '#{qual_name}'::regclass
-          and attnum > 0
-          and not attisdropped
-        order by attnum
-      )).map(&:to_sym)
-
-      values =
-          case data
-            when Hash; fields.map { |f| data[f] }
-            when OpenStruct; fields.map { |f| data.send(f) }
-            when Array; data
-          else
-            raise Error, "Illegal value #{data.inspect}"
-          end
-
-      "(#{quote_tuple(values, elem_types: elem_types)})::#{qual_name}"
+    def quote_record(data, schema_name = nil, type, elem_types: nil)
+      quote_record_impl(data, schema_name, type, elem_types: elem_types, array: false)
     end
 
-    def quote_records(schema_name = nil, type, datas, elem_types: nil)
-      qual_name = [schema_name, type].compact .join('.') + "[]"
-      records = datas.map { |data| quote_record(schema_name, type, data, elem_types: elem_types) }
-      "array[#{records.join(", ")}]::#{qual_name}"
+    # Quote an array of records. The type is the record type, not the type of
+    # the enclosing array
+    def quote_records(data, schema_name = nil, type, elem_types: nil)
+      quote_record_impl(data, schema_name, type, elem_types: elem_types, array: true)
     end
 
     # :call-seq:
@@ -980,6 +960,42 @@ module PgConn
       end
     end
 
+    # Common implementation for #quote_record and #quote_records that avoids
+    # query the database multiple times while not duplication the code. the
+    # :array flag controls the mode
+    def quote_record_impl(datas, schema_name = nil, type, elem_types: nil, array: nil)
+      pg_type = [schema_name, type].compact.join('.')
+      fields = self.values(%(
+        select attname
+        from pg_attribute
+        where
+          attrelid = '#{pg_type}'::regclass
+          and attnum > 0
+          and not attisdropped
+        order by attnum
+      )).map(&:to_sym)
+
+      datas = [datas] if !array
+
+      literals = datas.map { |data|
+        values =
+            case data
+              when Hash; fields.map { |f| data[f] }
+              when OpenStruct; fields.map { |f| data.send(f) }
+              when Array; data
+            else
+              raise Error, "Illegal value #{data.inspect}"
+            end
+        "(#{quote_tuple(values, elem_types: elem_types)})::#{pg_type}"
+      }
+
+      if array
+        "array[#{literals.join(', ')}]::#{pg_type}[]"
+      else
+        literals.first
+      end
+    end
+
     # :call-seq:
     #   pg_exec(string)
     #   pg_exec(array)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_conn
 version: !ruby/object:Gem::Version
-  version: 0.26.0
+  version: 0.26.1
 platform: ruby
 authors:
 - Claus Rasmussen
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-11-16 00:00:00.000000000 Z
+date: 2024-11-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg