Write to collection cache where the template is rendered.

Moves us closer to having access to a local template variable,
we can ask for eligibility and its virtual_path.

Currently we rely on `@template`, which we don't have available
when rendering collections without a fixed template.

1 files changed, 8 insertions, 9 deletions

diff --git a/actionview/lib/action_view/renderer/partial_renderer/collection_caching.rb b/actionview/lib/action_view/renderer/partial_renderer/collection_caching.rb
index c353eb0b31..a037424498 100644
--- a/actionview/lib/action_view/renderer/partial_renderer/collection_caching.rb
+++ b/actionview/lib/action_view/renderer/partial_renderer/collection_caching.rb
@@ -51,16 +51,15 @@ module ActionView
       end
 
       def fetch_or_cache_partial(cached_partials, order_by:)
-        rely_on_individual_cache_call = !callable_cache_key?
-
         order_by.map do |key|
-          cached_partials.fetch(key) do
-            yield.tap do |rendered_partial|
-              unless rely_on_individual_cache_call
-                collection_cache.write(key, rendered_partial, @options[:cache_options])
-              end
-            end
-          end
+          cached_partials.fetch(key) { yield }
+        end
+      end
+
+      def collection_cache_rendered_partial(rendered_partial, key_by)
+        if callable_cache_key?
+          key = expanded_cache_key(@options[:cache].call(key_by))
+          collection_cache.write(key, rendered_partial, @options[:cache_options])
         end
       end
   end