pg_conn 0.16.0 → 0.17.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 25c23025d55818536e6d89bc5fa40b12ac296f0e61657db387f8a8732bdf3504
-  data.tar.gz: 7592d3a2b26d7f75f0ea88ad9b8f951422ced0cc913a387eb3f774b31bb1c4eb
+  metadata.gz: 64ed87e1cf308dd73256ca1a3785acee059ffbd8fc30d11303f0b494799b9dcb
+  data.tar.gz: 21076b277ee98f430d9fa19a882ab4a9242815b749a956ec76a210464517162c
 SHA512:
-  metadata.gz: 0701ef07551bc1db06b654aa12e957ca4ad2286f1050c122af95ce78670ea426e6bf8f566831070200588e2c844eaf70ddab0cc5147941ccdcf6f4dc232c4102
-  data.tar.gz: 5310090682a9c93a648823b477081eebe31c214554ebe06a44d2f6702ec18c3fd5eb43e16bad0c15178d48a6194e109dbf15adb8bb3f75e645b90581ef3bc948
+  metadata.gz: f657154a19e12be06b0e98668964f3b157745f3bbd90bad5e242416598e4be887c89dc7ae81ec56a109bf51aa66997e563825c43fb6a73d09e09711360a2395b
+  data.tar.gz: 3b5892c3e8f80dcae50cffef9ea128dab6d1bb349d72115c9cb2b5ceff3f0371c1622851192b78ecea8dc572812af6aa9e1b9133ef119b31f81f0772f7b6239a

data/lib/pg_conn/schema_methods.rb

@@ -124,6 +124,45 @@ module PgConn
       raise NotImplementedError
     end
 
+    # Return name of the table's sequence (if any)
+    def sequence(schema, table) 
+      conn.value "select pg_get_serial_sequence('#{schema}.#{table}', 'id')" 
+    end
+
+    # Get the current serial value for the table. Returns nil if the serial has
+    # not been used. If :next is true, the next value will be returned
+    def get_serial(schema, table, next: false)
+      uid = "#{schema}.#{table}"
+      next_option = binding.local_variable_get(:next) # because 'next' is a keyword
+
+      seq = sequence(schema, table) or raise ArgumentError, "Table #{uid} does not have a sequence"
+      value = conn.value %(
+        select
+          case is_called
+            when true then last_value
+            else null
+          end as "value"
+        from
+          #{seq}
+      )
+      if next_option
+        value&.+(1) || 1
+      else
+        value
+      end
+    end
+
+    # Set the serial value for the table
+    def set_serial(schema, table, value)
+      uid = "#{schema}.#{table}"
+      seq = sequence(schema, table) or raise ArgumentError, "Table #{uid} does not have a sequence"
+      if value
+        conn.exec "select setval('#{seq}', #{value})"
+      else
+        conn.exec "select setval('#{seq}', 1, false)"
+      end
+    end
+
   private
     def relation_exist_query(schema, relation, kind: nil)
       kind_sql_list = "'" + (kind.nil? ? %w(r f v m) : Array(kind).flatten).join("', '") + "'"

data/lib/pg_conn/version.rb

@@ -1,3 +1,3 @@
 module PgConn
-  VERSION = "0.16.0"
+  VERSION = "0.17.1"
 end

data/lib/pg_conn.rb

@@ -558,6 +558,7 @@ module PgConn
 
     # Insert record(s) in a table. Returns the id of the inserted record(s)
     def insert(schema = nil, table, array_or_hash)
+      return [] if array_or_hash.empty?
       table = [schema, table].compact.join(".")
       is_array = array_or_hash.is_a?(Array)
       array = is_array ? array_or_hash : [array_or_hash]

metadata

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