module ActiveRecord::ConnectionAdapters::AbstractPool

def get_schema_cache(connection)
  self.schema_cache ||= SchemaCache.new(connection)
  schema_cache.connection = connection
  schema_cache
end

def lazily_set_schema_cache
  return unless ActiveRecord.lazily_load_schema_cache
  cache = SchemaCache.load_from(db_config.lazy_schema_cache_path)
  set_schema_cache(cache)
end

def set_schema_cache(cache)
  self.schema_cache = cache
end