application_seeds 0.5.0 → 0.6.0

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+v0.6.0, 2014-03-06
+------------------
+
+  * Added method that will allow the user to defer the enforcement of
+    referential integrity checks while a block of code is being
+    executed.
+
+
 v0.5.0, 2014-02-28
 ------------------
 

data/README.md

@@ -363,6 +363,16 @@ ApplicationSeeds.config_value(:num_departments)
 => 3
 ```
 
+And use them in your seed files:
+
+```ruby
+<% ApplicationSeeds.config_value(:num_companies).times do |x| %>
+company_<%= x %>:
+  name: Company_<%= x %>
+<% end %>
+```
+
+
 ### Merging config value files
 
 If you are using nested datasets, then all of the appropriate
@@ -554,6 +564,19 @@ are unable to insert new data into the databse after your dataset has
 been imported, then this should correct them.
 
 
+### Defer referential integrity checks
+
+```ruby
+ApplicationSeeds.defer_referential_integrity_checks do
+  # Process some seed data
+end
+```
+
+This method will defer the enforcement of foreign key contraints while
+the block of code is being executed.  This is useful when creating
+chunks of seed data that have are dependent on each other's existance.
+
+
 ### Fetch data from the `_config.yml` files
 
 ```ruby

data/lib/application_seeds.rb

@@ -158,6 +158,15 @@ module ApplicationSeeds
       end
     end
 
+    #
+    # Defer the enforcement of foreign key constraints while the block is being executed.
+    #
+    def defer_referential_integrity_checks
+      Database.without_foreign_keys do
+        yield
+      end
+    end
+
     private
 
     def dataset_path(dataset)

data/lib/application_seeds/database.rb

@@ -1,7 +1,7 @@
 module ApplicationSeeds
   class Database
-
     class << self
+
       def connection
         return @connection unless @connection.nil?
 
@@ -21,8 +21,43 @@ module ApplicationSeeds
         connection.exec('DROP TABLE IF EXISTS application_seeds;')
         connection.exec('CREATE TABLE application_seeds (dataset varchar(255));')
       end
-    end
 
+      def without_foreign_keys
+        drop_foreign_keys_sql = generate_drop_foreign_keys_sql
+        create_foreign_keys_sql = generate_create_foreign_keys_sql
+
+        connection.exec(drop_foreign_keys_sql)
+        yield
+        connection.exec(create_foreign_keys_sql)
+      end
+
+      private
+
+      def generate_drop_foreign_keys_sql
+        result = connection.exec <<-SQL
+          SELECT 'ALTER TABLE '||nspname||'.'||relname||' DROP CONSTRAINT '||conname||';'
+          FROM pg_constraint
+          INNER JOIN pg_class ON conrelid=pg_class.oid
+          INNER JOIN pg_namespace ON pg_namespace.oid=pg_class.relnamespace
+          WHERE contype='f'
+          ORDER BY CASE WHEN contype='f' THEN 0 ELSE 1 END,contype,nspname,relname,conname
+        SQL
+        result.values.join
+      end
+
+      def generate_create_foreign_keys_sql
+        result = connection.exec <<-SQL
+          SELECT 'ALTER TABLE '||nspname||'.'||relname||' ADD CONSTRAINT '||conname||' '|| pg_get_constraintdef(pg_constraint.oid)||';'
+          FROM pg_constraint
+          INNER JOIN pg_class ON conrelid=pg_class.oid
+          INNER JOIN pg_namespace ON pg_namespace.oid=pg_class.relnamespace
+          WHERE contype='f'
+          ORDER BY CASE WHEN contype='f' THEN 0 ELSE 1 END DESC,contype DESC,nspname DESC,relname DESC,conname DESC;
+        SQL
+        result.values.join
+      end
+
+    end
   end
 end
 

data/lib/application_seeds/version.rb

@@ -1,3 +1,3 @@
 module ApplicationSeeds
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: application_seeds
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-28 00:00:00.000000000 Z
+date: 2014-03-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -123,7 +123,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 632668571274566106
+      hash: -1007490013987030043
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -132,7 +132,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 632668571274566106
+      hash: -1007490013987030043
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.23