pg_conn 0.16.0 → 0.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 25c23025d55818536e6d89bc5fa40b12ac296f0e61657db387f8a8732bdf3504
-  data.tar.gz: 7592d3a2b26d7f75f0ea88ad9b8f951422ced0cc913a387eb3f774b31bb1c4eb
+  metadata.gz: b6f8840c223cb2dd9a6a5f8b2211e43184eb308114ffdd8bdc76d92f6ea03e16
+  data.tar.gz: fbfad4033c2d21abb91b726c183001d4edbaf348d61cb425b4b003dd89a37f35
 SHA512:
-  metadata.gz: 0701ef07551bc1db06b654aa12e957ca4ad2286f1050c122af95ce78670ea426e6bf8f566831070200588e2c844eaf70ddab0cc5147941ccdcf6f4dc232c4102
-  data.tar.gz: 5310090682a9c93a648823b477081eebe31c214554ebe06a44d2f6702ec18c3fd5eb43e16bad0c15178d48a6194e109dbf15adb8bb3f75e645b90581ef3bc948
+  metadata.gz: fe6647045b3bad635ed480e201967a20f0b11b5fd71bcbdec6766b6a2f056e3b0c8de045e87f56af90fb429130b2a9f105c46f5de5de31219b8384d18b27b3ca
+  data.tar.gz: 5e114bd53908faf4a881799035276151ace147c63156cc006359e6483c182d0522d2a724f4e7b4fd551914739b23240c06a55319b46d4aa68c6bf65f0f8da095

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.0"
 end

metadata

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