39 lines
		
	
	
	
		
			1.5 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			39 lines
		
	
	
	
		
			1.5 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
# frozen_string_literal: true
 | 
						|
 | 
						|
namespace :db do
 | 
						|
  namespace :migrate do
 | 
						|
    desc 'Setup the db or migrate depending on state of db'
 | 
						|
    task setup: :environment do
 | 
						|
      begin
 | 
						|
        if ActiveRecord::Migrator.current_version.zero?
 | 
						|
          Rake::Task['db:migrate'].invoke
 | 
						|
          Rake::Task['db:seed'].invoke
 | 
						|
        end
 | 
						|
      rescue ActiveRecord::NoDatabaseError
 | 
						|
        Rake::Task['db:setup'].invoke
 | 
						|
      else
 | 
						|
        Rake::Task['db:migrate'].invoke
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  task :post_migration_hook do
 | 
						|
    at_exit do
 | 
						|
      unless %w(C POSIX).include?(ActiveRecord::Base.connection.select_one('SELECT datcollate FROM pg_database WHERE datname = current_database();')['datcollate'])
 | 
						|
        warn <<~WARNING
 | 
						|
          Your database collation may be susceptible to index corruption.
 | 
						|
            (This warning does not indicate that index corruption has occurred, and it can be ignored if you've previously checked for index corruption)
 | 
						|
            (To learn more, visit: https://docs.joinmastodon.org/admin/troubleshooting/index-corruption/)
 | 
						|
        WARNING
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  task :pre_migration_check do
 | 
						|
    version = ActiveRecord::Base.connection.select_one("SELECT current_setting('server_version_num') AS v")['v'].to_i
 | 
						|
    abort 'ERROR: This version of Mastodon requires PostgreSQL 9.5 or newer. Please update PostgreSQL before updating Mastodon.' if version < 90_500
 | 
						|
  end
 | 
						|
 | 
						|
  Rake::Task['db:migrate'].enhance(['db:pre_migration_check'])
 | 
						|
  Rake::Task['db:migrate'].enhance(['db:post_migration_hook'])
 | 
						|
end
 |