diff options
| author | Gary S. Weaver <garysweaver@gmail.com> | 2012-12-18 10:46:16 -0500 |
|---|---|---|
| committer | Gary S. Weaver <garysweaver@gmail.com> | 2012-12-18 10:46:16 -0500 |
| commit | 9bb27f7ffe3eb732df737e477cd8fc25e007f77b (patch) | |
| tree | a59deb84ea5ca4fb4ed257d35afe47eb64f04858 /activerecord/lib/active_record | |
| parent | 15c40b2ef51120974edc626106de69faa53dfc3f (diff) | |
| download | rails-9bb27f7ffe3eb732df737e477cd8fc25e007f77b.tar.gz rails-9bb27f7ffe3eb732df737e477cd8fc25e007f77b.tar.bz2 rails-9bb27f7ffe3eb732df737e477cd8fc25e007f77b.zip | |
#5523 Add ability for postgresql adapter to disable user triggers in disable_referential_integrity.
Diffstat (limited to 'activerecord/lib/active_record')
| -rw-r--r-- | activerecord/lib/active_record/connection_adapters/postgresql/referential_integrity.rb | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/activerecord/lib/active_record/connection_adapters/postgresql/referential_integrity.rb b/activerecord/lib/active_record/connection_adapters/postgresql/referential_integrity.rb index 16da3ea732..bc775394a6 100644 --- a/activerecord/lib/active_record/connection_adapters/postgresql/referential_integrity.rb +++ b/activerecord/lib/active_record/connection_adapters/postgresql/referential_integrity.rb @@ -7,13 +7,21 @@ module ActiveRecord end def disable_referential_integrity #:nodoc: - if supports_disable_referential_integrity? then - execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL" }.join(";")) + if supports_disable_referential_integrity? + begin + execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER ALL" }.join(";")) + rescue + execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} DISABLE TRIGGER USER" }.join(";")) + end end yield ensure - if supports_disable_referential_integrity? then - execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL" }.join(";")) + if supports_disable_referential_integrity? + begin + execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER ALL" }.join(";")) + rescue + execute(tables.collect { |name| "ALTER TABLE #{quote_table_name(name)} ENABLE TRIGGER USER" }.join(";")) + end end end end |