table_saw 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1339fbe2ad8b61d13d84595da082e4bd8886f9ecb56e1c8ce55ba9b5d9d33304
-  data.tar.gz: 79e448b1ee7d8f77a019b491ed5cbe11e4b5ca353721643211543813a779b856
+  metadata.gz: fb4c657ec087524b5db0ab9619be50b819d5b65ba1f410bfcc20b04cebc55123
+  data.tar.gz: a82d41d12074742fa1a120322dca716d378d8ee1a5e78ca30820176f2e44b37a
 SHA512:
-  metadata.gz: 49bf13e6be2584987b29755ca760fced988ff01f7f8ad362a2cb35122f83da8de7b1de37c0f02d32bee9dee733b0d9c517d515dc5134cb00a4e08c26f57af9c1
-  data.tar.gz: 290706ba80434334c9d4af235e84972018bad3d2ef009eb85ee288ce6e7739fb669e1ba428e8060e6c1509910a0cdb0bffe05f26fdeee562772b69d41c1fe687
+  metadata.gz: 1069f4020c08827258a12e14433e6782c2824c5e58917251433636f6e8b5ca5a951d0f67edc582cc893a36b6f3642dce484bd76649f01bdab2e4d750496c4607
+  data.tar.gz: 4c7ff84820fdd86ebd72ceee70d22707bc39efe05264d7ee46888f2459b0433c769125e1d9142623265b02e300438d09d9ff53a92109b44e57e1ee55b7c0d81b

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    table_saw (0.4.0)
+    table_saw (0.5.0)
       connection_pool
       pg
       thor

data/lib/table_saw/create_dump_file.rb

@@ -11,6 +11,8 @@ module TableSaw
 
     # rubocop:disable Metrics/MethodLength,Metrics/AbcSize
     def call
+      File.delete(file) if File.exist?(file)
+
       write_to_file <<~SQL
         BEGIN;
 
@@ -67,9 +69,9 @@ module TableSaw
     end
 
     def restart_sequences
-      records.each_key do |table|
+      TableSaw::Queries::SerialSequences.new.call.slice(*records.keys).each do |table, sequence|
         write_to_file <<~SQL
-          select setval(pg_get_serial_sequence('#{table}', 'id'), (select max(id) from #{table}), true);
+          select setval('#{sequence.name}', (select max(#{sequence.column}) from #{table}), true);
         SQL
       end
 

data/lib/table_saw/queries/serial_sequences.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module TableSaw
+  module Queries
+    class SerialSequences
+      QUERY = <<~SQL
+        select
+          pg_get_serial_sequence(kcu.table_name, kcu.column_name) as sequence,
+          kcu.table_name as table,
+          kcu.column_name as column
+        from information_schema.key_column_usage as kcu
+                 inner join information_schema.table_constraints as tc
+                            on tc.constraint_name = kcu.constraint_name
+        where tc.constraint_type = 'PRIMARY KEY'
+          and pg_get_serial_sequence(kcu.table_name, kcu.column_name) is not null
+      SQL
+
+      SerialSequence = Struct.new(:name, :table, :column)
+
+      def call
+        TableSaw::Connection.with do |conn|
+          conn.exec(QUERY).each_with_object({}) do |row, memo|
+            memo[row['table']] = SerialSequence.new(row['sequence'], row['table'], row['column'])
+          end
+        end
+      end
+    end
+  end
+end

data/lib/table_saw/queries.rb

@@ -3,4 +3,5 @@
 require 'table_saw/queries/foreign_key_relationships'
 require 'table_saw/queries/materialized_views'
 require 'table_saw/queries/no_id_tables'
+require 'table_saw/queries/serial_sequences'
 require 'table_saw/queries/table_columns'

data/lib/table_saw/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module TableSaw
-  VERSION = '0.4.0'
+  VERSION = '0.5.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: table_saw
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Hamed Asghari
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-06-14 00:00:00.000000000 Z
+date: 2019-06-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: connection_pool
@@ -231,6 +231,7 @@ files:
 - lib/table_saw/queries/foreign_key_relationships.rb
 - lib/table_saw/queries/materialized_views.rb
 - lib/table_saw/queries/no_id_tables.rb
+- lib/table_saw/queries/serial_sequences.rb
 - lib/table_saw/queries/table_columns.rb
 - lib/table_saw/version.rb
 - table_saw.gemspec