internetee-registry/lib/tasks/whois.rake

namespace :whois do
  desc 'Regenerate Registry whois_records table and sync with whois server (slower)'
  task regenerate: :environment do
    start = Time.zone.now.to_f

    @i = 0
    print "-----> Regenerate Registry whois_records table and sync with whois server..."
    ActiveRecord::Base.uncached do
      puts "\n#{@i}"
      Domain.included.find_in_batches(batch_size: 10000) do |batch|
        batch.map(&:update_whois_record)
        puts(@i += 10000)
        GC.start
      end
    end
    puts "\n-----> all done in #{(Time.zone.now.to_f - start).round(2)} seconds"
  end

  desc 'Create whois database'
  task create: [:environment] do
    whois_db = "whois_#{Rails.env}"
    begin
      puts "\n------------------------ Create #{whois_db} ---------------------------------------\n"
      ActiveRecord::Base.clear_all_connections!
      conf = ActiveRecord::Base.configurations
      
      ActiveRecord::Base.connection.create_database(conf[whois_db]['database'].to_sym, conf[whois_db])
    rescue => e
      puts "\n#{e}"
    end
  end

  desc 'Drop whois database'
  task drop: [:environment] do
    # just in case we allow only drop test, comment it out only for temp
    if Rails.env.test?
      whois_db = "whois_#{Rails.env}"

      begin
        puts "\n------------------------ #{whois_db} drop ------------------------------\n"
        ActiveRecord::Base.clear_all_connections!
        ActiveRecord::Base.establish_connection(whois_db.to_sym)

        conf = ActiveRecord::Base.configurations
        if ActiveRecord::Tasks::DatabaseTasks.drop(conf[whois_db])
          puts "#{conf[whois_db]['database']} dropped"
        else
          puts "Didn't find database #{whois_db}, no drop"
        end
      rescue => e
        puts "\n#{e}"
      end
    else
      puts 'Only for test'
    end
  end

  namespace :schema do
    desc 'Load whois schema into empty whois database'
    task load: [:environment] do
      whois_db = "whois_#{Rails.env}"
      begin
        puts "\n------------------------ #{whois_db} schema loading ------------------------------\n"
        ActiveRecord::Base.clear_all_connections!
        ActiveRecord::Base.establish_connection(whois_db.to_sym)
        if ActiveRecord::Base.connection.table_exists?('schema_migrations')
          puts 'Found tables, skip schema load!'
        else
          load("#{Rails.root}/db/#{schema_file(whois_db)}")
        end
      rescue => e
        puts "\n#{e}"
      end
    end
  end
end