module Bullet::ActiveRecord

def construct_model(record, node, row, model_cache, id, aliases)
  result = origin_construct_model(record, node, row, model_cache, id, aliases)
  associations = node.reflection.name
  Bullet::Detector::Association.add_object_associations(record, associations)
  Bullet::Detector::NPlusOneQuery.call_association(record, associations)
  @bullet_eager_loadings[record.class] ||= {}
  @bullet_eager_loadings[record.class][record] ||= Set.new
  @bullet_eager_loadings[record.class][record] << associations
  result
end