Forked from javier-menendez/sqlite3_disable_referential_to_rails_5.rb
Created
February 8, 2019 15:24
-
-
Save ultrawebmarketing/08f84abdf100b89b6eef03eeb77877f8 to your computer and use it in GitHub Desktop.
Rails 5 initializer for disable foreign keys during `alter_table` for sqlite3 adapter
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
require 'active_record/connection_adapters/sqlite3_adapter' | |
# | |
# Monkey-patch for disable foreign keys during `alter_table` for sqlite3 adapter for Rails 5 | |
# | |
module ActiveRecord | |
module ConnectionAdapters | |
class SQLite3Adapter < AbstractAdapter | |
# REFERENTIAL INTEGRITY ==================================== | |
def disable_referential_integrity # :nodoc: | |
old_foreign_keys = query_value("PRAGMA foreign_keys") | |
old_defer_foreign_keys = query_value("PRAGMA defer_foreign_keys") | |
begin | |
execute("PRAGMA defer_foreign_keys = ON") | |
execute("PRAGMA foreign_keys = OFF") | |
yield | |
ensure | |
execute("PRAGMA defer_foreign_keys = #{old_defer_foreign_keys}") | |
execute("PRAGMA foreign_keys = #{old_foreign_keys}") | |
end | |
end | |
def insert_fixtures_set(fixture_set, tables_to_delete = []) | |
disable_referential_integrity do | |
transaction(requires_new: true) do | |
tables_to_delete.each {|table| delete "DELETE FROM #{quote_table_name(table)}", "Fixture Delete"} | |
fixture_set.each do |table_name, rows| | |
rows.each {|row| insert_fixture(row, table_name)} | |
end | |
end | |
end | |
end | |
private | |
def alter_table(table_name, options = {}) | |
altered_table_name = "a#{table_name}" | |
caller = lambda {|definition| yield definition if block_given?} | |
transaction do | |
disable_referential_integrity do | |
move_table(table_name, altered_table_name, | |
options.merge(temporary: true)) | |
move_table(altered_table_name, table_name, &caller) | |
end | |
end | |
end | |
end | |
end | |
end | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment