table_saw 1.3.0 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ce0e71ab653f2467be613e32219a81dbd1c8544b0ffe54dcd300075ef9d85d5f
-  data.tar.gz: 6758d8eb4b87c61e99425abc727d2e311c84d809ab1ab365cc3ec1c1d44dddfb
+  metadata.gz: f87bd795a89acabf42bd2ff9cc75dd2035d024cc877d8cf47f45cf8cc96b3029
+  data.tar.gz: dce4ea88bed8c8cf92e68020f528721dbe43eebcfa9adb92f3e0669a90cba6f3
 SHA512:
-  metadata.gz: 82c0db05a5e5d2cfcbd6141e78652e84380f65d960ab130b427ebe1da3caefc8f90b5b9f45621620d2d4a204f8b1fded43bc10978bbbec2f551442fddb5f0f4a
-  data.tar.gz: 748b769670ad9b24abe6a84b402cbbb2e1325e030d180ae9461b299b78af2b1b23611fa06d1a281b4a9cfac43f9842685ff1276d458941200204f47c5107b1cf
+  metadata.gz: c9f263214ec9a1d04f7e1c795c7d91b8a476cd9776ce67d16e4bec86a1db93684c9183090ff7cf7926bf29fe13260abafb24f56c71c59d5500e991983f24db61
+  data.tar.gz: 2274e86e0cdc654b5635400910c532957db83dc7f7de14f3ee497f8e7fecf31fd7d6dea1c9ec12a9fc2eeb8794bf6e816de9499f37ae1acb4af95e38c4c91443

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    table_saw (1.3.0)
+    table_saw (2.0.0)
       pg
       thor
 
@@ -41,7 +41,7 @@ GEM
       railties (>= 3.0.0)
       thor (>= 0.14.6)
     concurrent-ruby (1.1.5)
-    crass (1.0.4)
+    crass (1.0.5)
     database_cleaner (1.7.0)
     diff-lcs (1.3)
     docile (1.3.1)
@@ -50,13 +50,13 @@ GEM
       concurrent-ruby (~> 1.0)
     jaro_winkler (1.5.2)
     json (2.2.0)
-    loofah (2.2.3)
+    loofah (2.3.1)
       crass (~> 1.0.2)
       nokogiri (>= 1.5.9)
     method_source (0.9.2)
     mini_portile2 (2.4.0)
     minitest (5.11.3)
-    nokogiri (1.10.3)
+    nokogiri (1.10.5)
       mini_portile2 (~> 2.4.0)
     parallel (1.17.0)
     parser (2.6.3.0)

data/lib/table_saw/create_dump_file.rb

@@ -13,10 +13,11 @@ module TableSaw
     def call
       File.delete(file) if File.exist?(file)
 
+      alter_constraints_to_deferrable
+
       write_to_file <<~SQL
         BEGIN;
 
-        SET session_replication_role = replica;
         SET statement_timeout = 0;
         SET lock_timeout = 0;
         SET client_encoding = 'UTF8';
@@ -28,6 +29,8 @@ module TableSaw
       SQL
 
       records.each do |name, table|
+        defer_constraints(name)
+
         write_to_file <<~COMMENT
           --
           -- Data for Name: #{name}; Type: TABLE DATA
@@ -60,6 +63,34 @@ module TableSaw
 
     private
 
+    def alter_constraints_to_deferrable
+      records.each_key do |name|
+        write_to_file <<~COMMENT
+          --
+          -- Alter Constraints for Name: #{name}; Type: DEFERRABLE
+          --
+
+        COMMENT
+
+        TableSaw.information_schema.constraint_names[name].each do |constraint_name|
+          write_to_file "ALTER TABLE #{name} ALTER CONSTRAINT #{constraint_name} DEFERRABLE;"
+        end
+      end
+    end
+
+    def defer_constraints(name)
+      write_to_file <<~COMMENT
+        --
+        -- Set Constraints for Name: #{name}; Type: DEFERRED
+        --
+
+      COMMENT
+
+      TableSaw.information_schema.constraint_names[name].each do |constraint_name|
+        write_to_file "SET CONSTRAINTS #{constraint_name} DEFERRED;"
+      end
+    end
+
     def refresh_materialized_views
       TableSaw::Queries::MaterializedViews.new.call.each do |view|
         write_to_file "refresh materialized view #{view};"

data/lib/table_saw/information_schema.rb

@@ -2,12 +2,10 @@
 
 module TableSaw
   class InformationSchema
-    def belongs_to
-      foreign_key_relationships.belongs_to
-    end
-
-    def has_many
-      foreign_key_relationships.has_many
+    %i(belongs_to constraint_names has_many).each do |method_name|
+      define_method method_name do
+        foreign_key_relationships.public_send method_name
+      end
     end
 
     def primary_keys

data/lib/table_saw/queries/foreign_key_relationships.rb

@@ -5,6 +5,7 @@ module TableSaw
     class ForeignKeyRelationships
       QUERY = <<~SQL
         select
+          tc.constraint_name,
           tc.table_name as from_table,
           kcu.column_name as from_column,
           ccu.table_name as to_table,
@@ -27,6 +28,12 @@ module TableSaw
         end
       end
 
+      def constraint_names
+        @constraint_names ||= result.each_with_object(Hash.new { |h, k| h[k] = [] }) do |row, memo|
+          memo[row['from_table']].push(row['constraint_name'])
+        end
+      end
+
       private
 
       def result

data/lib/table_saw/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: table_saw
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 2.0.0
 platform: ruby
 authors:
 - Hamed Asghari
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-08-15 00:00:00.000000000 Z
+date: 2019-11-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg