BathHacked/energy-sparks

module Database
  class VacuumService
    def initialize(tables)
      @tables = tables
    end

    ## NB: VACUUM does not work when run inside a transaction block
    ## For rspec, use ts: false as an argument to the block to prevent tests being wrapped in a transaction block
    def perform(vacuum: false)
      @tables.each do |table|
        begin
          sql = vacuum ? "VACUUM ANALYSE #{table}" : "ANALYSE #{table}"
          ActiveRecord::Base.connection.execute(sql)
        rescue StandardError => exception
          message = "#{sql} error: #{exception.message}"
          Rails.logger.error(message)
          Rollbar.error(message)
        end
      end
    end
  end
end