Module: JitPreloader::ActiveRecordAssociationsCollectionAssociation

Defined in:
lib/jit_preloader/active_record/associations/collection_association.rb

Instance Method Summary collapse

Instance Method Details

#load_target ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/jit_preloader/active_record/associations/collection_association.rb', line 4

def load_target
  was_loaded = loaded?

  if !loaded? && owner.persisted? && owner.jit_preloader && (reflection.scope.nil? || reflection.scope.arity == 0)
    owner.jit_preloader.jit_preload(reflection.name)
  end

  jit_loaded = loaded?

  super.tap do |records|
    # We should not act on non-persisted objects, or ones that are already loaded.
    if owner.persisted? && !was_loaded
      # If we went through a JIT preload, then we will have attached another JitPreloader elsewhere.
      JitPreloader::Preloader.attach(records) if records.any? && !jit_loaded && JitPreloader.globally_enabled?

      # If the records were not pre_loaded
      records.each{ |record| record.jit_n_plus_one_tracking = true }

      if !jit_loaded && owner.jit_n_plus_one_tracking
        ActiveSupport::Notifications.publish("n_plus_one_query",
                                             source: owner, association: reflection.name)
      end
    end
  end
end