module Bullet::ActiveRecord

def construct_association(record, join, row)
  result = origin_construct_association(record, join, row)
  if Bullet.start?
    associations = join.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
  end
  result
end