Last active
October 11, 2023 00:56
-
-
Save pvcarrera/123280c58eca51ccebe3 to your computer and use it in GitHub Desktop.
Load Rails fixtures ignoring PostgreSQL constraints
This file contains 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
namespace :db do | |
desc 'Reset and populate sample data' | |
task populate_sample_data: [:reset, :drop_constraints, 'fixtures:load', :create_constraints] do | |
puts 'Fixtures loaded in the database' | |
end | |
desc 'Remove all database constraints' | |
task drop_constraints: [:environment, :create_constraints_script, :drop_constraints_script] do | |
ActiveRecord::Base.connection.execute(IO.read('tmp/drop_constraints.sql')) | |
puts 'Constraints dropped' | |
end | |
desc 'Recreate database constraints' | |
task create_constraints: [:environment, :create_constraints_script, :drop_constraints_script] do | |
ActiveRecord::Base.connection.execute(IO.read('tmp/create_constraints.sql')) | |
puts 'Constraints recreated' | |
end | |
file drop_constraints_script: :environment do | |
query = (<<-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 | |
ORDER BY CASE WHEN contype='f' THEN 0 ELSE 1 END,contype,nspname,relname,conname | |
SQL | |
array_to_file(ActiveRecord::Base.connection.execute(query), 'drop_constraints.sql') | |
end | |
file create_constraints_script: :environment do | |
query = (<<-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 | |
ORDER BY CASE WHEN contype='f' THEN 0 ELSE 1 END DESC,contype DESC,nspname DESC,relname DESC,conname DESC | |
SQL | |
array_to_file(ActiveRecord::Base.connection.execute(query), 'create_constraints.sql') | |
end | |
def array_to_file(array, file_name) | |
Dir.mkdir('tmp') unless File.exist?('tmp') | |
File.open("tmp/#{file_name}", 'w+') do |f| | |
f.puts(array.values.flatten) | |
end | |
end | |
end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment