pg_conn 0.44.0 → 0.46.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 51ccb11b750c683d57a02f274b4479366eced6d46c8908fed1e47aca302ff367
-  data.tar.gz: ad97bab6ef9fb0e7f821088a58772252ba72c4c9b695703b6ccd3a32fb1f5f80
+  metadata.gz: bc9f98c5804eb386c76e607d056c12ead0e44feb0df8dcabb85b3c057d824868
+  data.tar.gz: d0318211117d792acb5936b6d2b10557548fe491a030898cfc7ed55dfd901ad5
 SHA512:
-  metadata.gz: 629d0d6afb690bc1ef2f3490d395f81cf5177f86803f5e6049621fec73bcef07ed4b4c644cbf8e834059c7219076f2c62b1ad3b36e7fa1db344a415c9073213c
-  data.tar.gz: 50cbacb421a2e4f4332c53737d0b9e8cf88e2b6555fe3c814d728d957f5d1477065f4a5089ff7c11ca77c33634aabf8440bd6aebccee8b92e7c509b7a18ba3ef
+  metadata.gz: 6e2845bab10f1ff11373d43dfda7ad4650c38eff52fc6f874450fd4a9a294bd290be36d9922260ae2d1429bb218f62116b4a15eb521bd48ceaaa4381ac9685e1
+  data.tar.gz: c9d413eabf917f15c64dfd2165229203ebc3ca1dbef32ea42858934812dffd1ce257abe821c9a58fa864feaa26348fc76c86980599dad9287f0fc5aa44660cb6

data/lib/pg_conn/version.rb

@@ -1,3 +1,3 @@
 module PgConn
-  VERSION = "0.44.0"
+  VERSION = "0.46.0"
 end

data/lib/pg_conn.rb

@@ -498,7 +498,7 @@ module PgConn
       quote_record_impl(data, schema_name, type, array: true, **opts)
     end
 
-    # Return current search path. Note that a search path is part of
+    # Return current search path. Note that the search path is part of
     # the transaction
     def search_path
       self.value("show search_path").split(/,\s*/) - %w("$user" pg_temp)
@@ -832,11 +832,11 @@ module PgConn
     end
 
     # :call-seq:
-    #   insert(table, record|records)
-    #   insert(table, fields, record|records|tuples|values)
+    #   insert(table, struct|structs|record|records)
+    #   insert(table, fields, struct|structs|record|records|tuples|values)
     #
-    #   insert(schema, table, record|records)
-    #   insert(schema, table, fields, record|records|tuples|values)
+    #   insert(schema, table, struct|structs|record|records)
+    #   insert(schema, table, fields, struct|structs|record|records|tuples|values)
     #
     # Insert record(s) in table and return id(s)
     #
@@ -874,12 +874,15 @@ module PgConn
         method = :values # The pg_conn method when multiple records are inserted
         if data.empty?
           return []
-        elsif data.first.is_a?(Array) # Tuple (array) element. Requires the 'fields' argument
+        elsif data.first.is_a?(Array) # Tuple (Array) element. Requires the 'fields' argument
           fields or raise ArgumentError
           tuples = data
-        elsif data.first.is_a?(Hash) # Hash element
+        elsif data.first.is_a?(Hash) # Record (Hash) element
           fields ||= data.first.keys
           tuples = data.map { |record| fields.map { |field| record[field] } }
+        elsif data.first.is_a?(OpenStruct)
+          fields ||= data.first.to_h.keys
+          tuples = data.map { |struct| hash = struct.to_h; fields.map { |field| hash[field] } }
         else
           fields.size == 1 or raise ArgumentError, "Illegal number of fields, expected exactly one"
           tuples = data.map { |e| [e] }
@@ -888,6 +891,11 @@ module PgConn
         method = upsert ? :value? : :value # The pg_conn method when only one record is inserted
         fields ||= data.keys
         tuples = [fields.map { |field| data[field] }]
+      elsif data.is_a?(OpenStruct)
+        method = upsert ? :value? : :value # The pg_conn method when only one record is inserted
+        hash = data.to_h
+        fields ||= hash.keys
+        tuples = [fields.map { |field| hash[field] }]
       else
         raise ArgumentError, "Illegal argument '#{data.inspect}'"
       end
@@ -931,7 +939,7 @@ module PgConn
       exec %(update #{table} set #{assignments} where #{constraint})
     end
 
-    # Delete record(s)
+    # Delete record(s). See also #truncate
     def delete(schema = nil, table, expr)
       table = [schema, table].compact.join(".")
       constraint =
@@ -945,53 +953,21 @@ module PgConn
       exec %(delete from #{table} where #{constraint})
     end
 
-    # Execute block with the given set of global or local options and reset
-    # them afterwards
-    #
-    # Global options are :silent, :notice and :warning, they're very useful in
-    # RSpec tests
+    # :call-seq:
+    #   truncate(qual_table_name...)
+    #   truncate(schema, table_name...)
     #
-    # Local options are :search_path that runs the block with the given
-    # schemas, :username that runs the block as the given user, :commit that
-    # runs the block in a transaction if true or false; true commits the
-    # transaction and false rolls it back (very rarely useful). It is not run
-    # in a transaction if :commit is nil. :log controls logging like
-    # #logger= but nil (the default) is a nop
-    def with(**options, &block)
-      search_path = options.delete(:search_path)
-      username = options.delete(:username)
-      log = options.delete(:log)
-      commit = options.delete(:commit)
-
-      saved_options = @options.dup
-      saved_search_path = self.search_path if search_path
-      saved_logger = self.logger if log
-
-      begin
-        set_options(options)
-        self.search_path = search_path if search_path
-        self.logger = log if !log.nil?
-
-        inner = lambda {
-          if !commit.nil?
-            self.transaction(commit: commit) {
-              block.yield
-            }
-          else
-            block.yield
-          end
-        }
-
-        if username
-          self.su(username, &inner)
-        else
-          inner.call
-        end
-      ensure
-        self.logger = saved_logger if log
-        self.search_path = saved_search_path if search_path
-        set_options(saved_options)
+    # Empty a table using Sql 'truncate'. Arguments are flattened before processing
+    def truncate(*args)
+      args = args.flatten
+      if args.first =~ /\./
+        tables = args
+      else
+        schema = args.shift
+        !args.empty? or raise ArgumentError, "Table argument expected"
+        tables = args.map { |table| [schema, table].compact.join('.') }
       end
+      exec %(truncate #{quote_identifiers(tables)})
     end
 
     # Execute SQL statement(s) in a transaction and return the number of
@@ -1058,6 +1034,55 @@ module PgConn
       end
     end
 
+    # Execute block with the given set of global or local options and reset
+    # them afterwards
+    #
+    # Global options are :silent, :notice and :warning, they're very useful in
+    # RSpec tests
+    #
+    # Local options are :search_path that runs the block with the given
+    # schemas, :username that runs the block as the given user, :commit that
+    # runs the block in a transaction if true or false; true commits the
+    # transaction and false rolls it back (very rarely useful). It is not run
+    # in a transaction if :commit is nil. :log controls logging like
+    # #logger= but nil (the default) is a nop
+    def with(**options, &block)
+      search_path = options.delete(:search_path)
+      username = options.delete(:username)
+      log = options.delete(:log)
+      commit = options.delete(:commit)
+
+      saved_options = @options.dup
+      saved_search_path = self.search_path if search_path
+      saved_logger = self.logger if log
+
+      begin
+        set_options(options)
+        self.search_path = search_path if search_path
+        self.logger = log if !log.nil?
+
+        inner = lambda {
+          if !commit.nil?
+            self.transaction(commit: commit) {
+              block.yield
+            }
+          else
+            block.yield
+          end
+        }
+
+        if username
+          self.su(username, &inner)
+        else
+          inner.call
+        end
+      ensure
+        self.logger = saved_logger if log
+        self.search_path = saved_search_path if search_path
+        set_options(saved_options)
+      end
+    end
+
     # Switch user to the given user and execute the statement before swithcing
     # back to the original user
     #
@@ -1269,7 +1294,7 @@ module PgConn
     end
 
     # Common implementation for #quote_record and #quote_records that avoids
-    # querying the database multiple times or duplication the code
+    # querying the database multiple times or duplicating the code
     #
     # @data can be a Hash, Array, or OpenStruct. Hash keys must be symbols or
     # strings, they belong to the same namespace so :k and "k" refer to the

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_conn
 version: !ruby/object:Gem::Version
-  version: 0.44.0
+  version: 0.46.0
 platform: ruby
 authors:
 - Claus Rasmussen