ActiveRecord::ConnectionAdapters::AbstractAdapter.class_eval do
  def mark_active_table_as_loaded(table_name)
    @active_tables_loaded ||= []
    @active_tables_loaded << table_name
  end

  def active_table_loaded_for?(table_name)
    return false if @active_tables_loaded.blank?
    @active_tables_loaded.include?(table_name)
  end
end