n1_loader 2.0.1 → 2.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1ed46e8e8cbb342314dc3b81c510723aa2b010c927077c074ed487db90c762c2
-  data.tar.gz: 6a79ef2b754868d40b8afb4b32533613f381df639eafc4b255fb5cf464df30ee
+  metadata.gz: 56479004f7a3456664bac05e808268177dabffa616848b57efb734865fd5ec49
+  data.tar.gz: 2d65c334abe26c5eb9baeb17ebe186d201b6c362b60d628758defd696b3e3522
 SHA512:
-  metadata.gz: df36fc5c04279a5dd7bba45979a86ae8e8c3775c210a6fef524139ce9460df59c0d8a23533e2825e5bc42db4f129a71d97b345adbc29777afcbc61a06b0ab3a8
-  data.tar.gz: 34ea3a6340cbf790da5fe0245939b0d0cd999ef5e656860c222eda1ed64bc068958b7a765b53b6dff25c86d781af5e781c372dcf5755da74e8ad5ebe09dc84f0
+  metadata.gz: 9d83a5cc7742f23b0702f56fbcd2320f46e309731c7734df825406a657709259dae08e2e3ed78db322f815438e5e0db82cbe40799e02bb7d0f7cd469ad453585
+  data.tar.gz: 225fa5c982e60ceaae8d5a8cbd90ff6ecd8d54cff5105bc1338d45e6fb9969586af5ced3fa6b3a7c6948a453ee7bd7956a30f2972527719d32467116da48f7ce

data/lib/n1_loader/ar_lazy_preload/loader_patch.rb

@@ -7,16 +7,16 @@ module N1Loader
       attr_accessor :context_setup
 
       def loaded
-        return @loaded if @already_loaded && @already_context
+        return @loaded_by_identity if @already_loaded && @already_context
 
         super
 
         synchronize do
-          context_setup&.call(@loaded.values.flatten) unless @already_context
+          context_setup&.call(@loaded_by_identity.values.flatten) unless @already_context
         end
 
         @already_context = true
-        @loaded
+        @loaded_by_identity
       end
     end
   end

data/lib/n1_loader/core/loadable.rb

@@ -32,11 +32,20 @@ module N1Loader
       n1_loaders[name]
     end
 
+    def n1_bind_to(collection)
+      @n1_binding = collection
+    end
+
     def n1_loader_reload(name)
-      n1_loaders[name] = LoaderCollection.new(self.class.n1_loaders[name], [self])
+      elements = @n1_binding || [self]
+      collection = LoaderCollection.new(self.class.n1_loaders[name], elements)
+
+      @n1_binding&.each { |el| el.n1_loaders[name] = collection if el.respond_to?(:n1_loaders) }
+      n1_loaders[name] = collection
     end
 
     def n1_clear_cache
+      @n1_binding = nil
       self.class.n1_loaders.each_key do |name|
         n1_loaders[name] = nil
       end

data/lib/n1_loader/core/loader.rb

@@ -47,12 +47,16 @@ module N1Loader
     end
 
     def for(element)
-      if loaded.empty? && elements.any?
+      identity_loaded = loaded
+
+      if identity_loaded.empty? && elements.any?
         raise NotFilled, "Nothing was preloaded, perhaps you forgot to use fulfill method"
       end
-      raise NotLoaded, "The data was not preloaded for the given element" unless loaded.key?(element)
 
-      loaded[element]
+      return identity_loaded[element] if identity_loaded.key?(element)
+      return @loaded_by_value[element] if @loaded_by_value.key?(element)
+
+      raise NotLoaded, "The data was not preloaded for the given element"
     end
 
     def cache_key
@@ -103,34 +107,37 @@ module N1Loader
     end
 
     def fulfill(element, value)
-      @loaded[element] = value
+      @loaded_by_identity[element] = value
+      @loaded_by_value[element] = value
     end
 
-    def loaded
-      return @loaded if @already_loaded
+    def ensure_loaded
+      return if @already_loaded
 
-      synchronize do
-        non_thread_safe_loaded unless @already_loaded
-      end
+      synchronize { non_thread_safe_loaded unless @already_loaded }
+    end
 
-      @loaded
+    def loaded
+      ensure_loaded
+      @loaded_by_identity
     end
 
-    def non_thread_safe_loaded
-      return @loaded if @already_loaded
+    def non_thread_safe_loaded # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
+      return if @already_loaded
 
       check_arguments!
 
-      @loaded = {}
+      @loaded_by_identity = {}.compare_by_identity
+      @loaded_by_value = {}
 
       if respond_to?(:single) && elements.size == 1
         fulfill(elements.first, single(elements.first))
       elsif elements.any?
+        elements.each { |el| el.n1_bind_to(elements) if el.respond_to?(:n1_bind_to) }
         perform(elements)
       end
 
       @already_loaded = true
-      @loaded
     end
   end
 end

data/lib/n1_loader/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module N1Loader
-  VERSION = "2.0.1"
+  VERSION = "2.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: n1_loader
 version: !ruby/object:Gem::Version
-  version: 2.0.1
+  version: 2.2.0
 platform: ruby
 authors:
 - Evgeniy Demin
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-12-28 00:00:00.000000000 Z
+date: 2026-03-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mutex_m
@@ -150,7 +150,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '1.3'
-description: 
+description:
 email:
 - lawliet.djez@gmail.com
 executables: []
@@ -188,7 +188,7 @@ metadata:
   source_code_uri: https://github.com/djezzzl/n1_loader
   changelog_uri: https://github.com/djezzzl/n1_loader/master/CHANGELOG.md
   funding_uri: https://opencollective.com/n1_loader#support
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -203,8 +203,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.33
-signing_key: 
+rubygems_version: 3.4.19
+signing_key:
 specification_version: 4
 summary: Loader to solve N+1 issue for good.
 test_files: []