JRUBY-3186: disable referential integrity for PG (Ryan Bell)

jruby · Nov 26, 2008 · 87f33e5 · 87f33e5
1 parent 91babd5
commit 87f33e5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/History.txt b/History.txt
@@ -13,6 +13,7 @@
   from court3nay)
 - JRUBY-2715: Add create/drop database methods to Postgres (Peter Williams)
 - JRUBY-3183: Fix structure dump for Postgres (Ryan Bell)
+- JRUBY-3186: disable referential integrity for PG (Ryan Bell)
 - Authoritative repository now hosted at
   git://github.com/nicksieger/activerecord-jdbc-adapter.git; rubyforge
   svn trunk cleaned out.

diff --git a/lib/jdbc_adapter/jdbc_postgre.rb b/lib/jdbc_adapter/jdbc_postgre.rb
@@ -336,6 +336,13 @@ def quoted_date(value)
       value.strftime("%Y-%m-%d %H:%M:%S")
     end
 
+    def disable_referential_integrity(&block) #:nodoc:
+      execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL" }.join(";"))
+      yield
+    ensure
+      execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL" }.join(";"))
+    end
+
     def rename_table(name, new_name)
       execute "ALTER TABLE #{name} RENAME TO #{new_name}"
     end