pg_conn 0.13.0 → 0.13.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7c975d085fc3e3f58a72fbb13d0e5ce9428254969e8723c8af0d6577381f0f70
-  data.tar.gz: 6a786d3466d2c0661b22ee738212a3b8a37afca86a99dbde59773e5d47b7a5f5
+  metadata.gz: 5a1eb7cce679245ec285129507e3d56948c2239b930ea18a757dd90c2e293404
+  data.tar.gz: c60efa88bb84530b91a58fd296781052b500968010808f9fa2c3dca4873db772
 SHA512:
-  metadata.gz: 6710fbd23379b8e1b13bd41dda8ce206fcefca30c93f43f2be0f4b7e437d854d50c89aa429cb70f52901eb7d4b515a810012801adc334b2082307dcc4ca2aaa9
-  data.tar.gz: 838a84b8f67710abcee3381f3691e93fab21c9d07ec48efa1480faec5c4b7621a5904d1000f2a62666c998f30b499de06c74eada70fac6609ea601beabf7b976
+  metadata.gz: de7cea3519b5dbd3763c24681ed917ae525fb87e359bc17461e0c950cc70ea47d3c79e230840d878a14533b7921a55384bf17235d32020787e22c37b8fb5e16c
+  data.tar.gz: 6dde3d8a9ea145db3e28b2467df50333042f3243da9d7b741e1195bb2d17856577541ff8df09744cc94517f9e02c7e6de44eba52fb3ba0ca363d0103edc89d13

data/lib/pg_conn/version.rb

@@ -1,3 +1,3 @@
 module PgConn
-  VERSION = "0.13.0"
+  VERSION = "0.13.2"
 end

data/lib/pg_conn.rb

@@ -547,27 +547,43 @@ module PgConn
       end
     end
 
-    # Insert record(s) in a table
+    # Insert record(s) in a table. Returns the id of the inserted record(s)
     def insert(schema = nil, table, array_or_hash)
       table = [schema, table].compact.join(".")
-      array = array_or_hash.is_a?(Array) ? array_or_hash : [array_or_hash]
+      is_array = array_or_hash.is_a?(Array)
+      array = is_array ? array_or_hash : [array_or_hash]
       identifiers = quote_identifier_list(array.first.keys)
       literals = array.map { |hash| quote_literal_list(hash.values) }.join(", ")
-      exec %(insert into #{table} #{identifiers} values #{literals})
+      method = (is_array ? :values : :value)
+      self.send method, %(insert into #{table} #{identifiers} values #{literals} returning id)
     end
 
     # Update record(s)
     def update(schema = nil, table, expr, hash)
       table = [schema, table].compact.join(".")
       assignments = hash.map { |k,v| "#{k} = #{quote_literal(v)}" }.join(", ")
-      constraint = (expr.is_a?(String) ? expr : "id = #{quote_literal(expr)}")
+      constraint = 
+          case expr
+            when String; expr
+            when Integer; "id = #{quote_literal(expr)}"
+            when Array; "id in #{quote_literal_list(expr)}"
+          else
+            raise ArgumentError
+          end
       exec %(update #{table} set #{assignments} where #{constraint})
     end
 
     # Delete record(s)
     def delete(schema = nil, table, expr)
       table = [schema, table].compact.join(".")
-      constraint = (expr.is_a?(String) ? expr : "id = #{quote_literal(expr)}")
+      constraint = 
+          case expr
+            when String; expr
+            when Integer; "id = #{quote_literal(expr)}"
+            when Array; "id in #{quote_literal_list(expr)}"
+          else
+            raise ArgumentError
+          end
       exec %(delete from #{table} where #{constraint})
     end
 

metadata

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