57 lines
2.1 KiB
Ruby
57 lines
2.1 KiB
Ruby
namespace :db do
|
|
desc 'Backup remote PostgreSQL database'
|
|
task :backup do
|
|
db_url = ENV.fetch('REMOTE_DB_URL') { raise 'Please set REMOTE_DB_URL' }
|
|
uri = URI.parse(db_url)
|
|
|
|
remote_host = uri.host
|
|
remote_port = uri.port
|
|
remote_user = uri.user
|
|
remote_db = uri.path.delete_prefix('/')
|
|
password = uri.password
|
|
|
|
backup_dir = File.expand_path('backups')
|
|
FileUtils.mkdir_p(backup_dir)
|
|
backup_file = File.join(backup_dir, "#{Time.now.strftime('%Y%m%d_%H%M%S')}-prod-backup.tar")
|
|
|
|
cmd = %W[
|
|
pg_dump -h #{remote_host} -p #{remote_port} -U #{remote_user} -d #{remote_db} -F t
|
|
--no-owner --exclude-extension=timescaledb --exclude-extension=timescaledb_toolkit
|
|
].join(' ')
|
|
|
|
puts "Backing up remote database to #{backup_file}..."
|
|
system({ 'PGPASSWORD' => password }, "#{cmd} > #{backup_file}")
|
|
puts 'Backup completed!'
|
|
end
|
|
|
|
desc 'Restore PostgreSQL database from backup (use CLEAN=1 to drop and recreate first)'
|
|
task :restore, [:backup_file] => [:environment] do |_, args|
|
|
local_user = ENV.fetch('LOCAL_DB_USER', 'justin')
|
|
local_db = ENV.fetch('LOCAL_DB_NAME', 'hensei_dev')
|
|
|
|
# Use the specified backup file or find the most recent one
|
|
backup_dir = File.expand_path('backups')
|
|
backup_file = args[:backup_file] || Dir.glob("#{backup_dir}/*-prod-backup.tar").max
|
|
|
|
raise 'Backup file not found. Please specify a valid backup file.' unless backup_file && File.exist?(backup_file)
|
|
|
|
if ENV['CLEAN']
|
|
puts 'Dropping and recreating database...'
|
|
ActiveRecord::Base.connection.execute(<<~SQL)
|
|
SELECT pg_terminate_backend(pid)
|
|
FROM pg_stat_activity
|
|
WHERE datname = '#{local_db}' AND pid <> pg_backend_pid()
|
|
SQL
|
|
ActiveRecord::Base.connection.disconnect!
|
|
Rake::Task['db:drop'].invoke
|
|
Rake::Task['db:create'].invoke
|
|
end
|
|
|
|
puts "Restoring database from #{backup_file}..."
|
|
system("pg_restore --no-owner --role=#{local_user} --disable-triggers -U #{local_user} -d #{local_db} #{backup_file}")
|
|
puts 'Restore completed!'
|
|
end
|
|
|
|
desc 'Backup remote database and restore locally'
|
|
task backup_and_restore: %i[backup restore]
|
|
end
|