diff --git a/core/lib/rom/cache.rb b/core/lib/rom/cache.rb index d3c989b4c..cba0ae783 100644 --- a/core/lib/rom/cache.rb +++ b/core/lib/rom/cache.rb @@ -46,7 +46,7 @@ def inspect # @api private def initialize - @objects = Concurrent::Map.new + @objects = ::Concurrent::Map.new @namespaced = {} end diff --git a/core/lib/rom/container.rb b/core/lib/rom/container.rb index 70ebeb39f..91707d101 100644 --- a/core/lib/rom/container.rb +++ b/core/lib/rom/container.rb @@ -110,6 +110,16 @@ def self.new(gateways, relations, mappers, commands) end end + # @api private + attr_reader :cache + + # @api private + def initialize + super + + @cache = Cache.new + end + # Return registered gateways # # @return [HashGateway>] diff --git a/core/lib/rom/plugins/relation/registry_reader.rb b/core/lib/rom/plugins/relation/registry_reader.rb index 28c754f08..5a0602bca 100644 --- a/core/lib/rom/plugins/relation/registry_reader.rb +++ b/core/lib/rom/plugins/relation/registry_reader.rb @@ -14,13 +14,9 @@ class RegistryReader < ::Module EMPTY_REGISTRY = RelationRegistry.build(EMPTY_HASH).freeze # @api private - attr_reader :relations - - # @api private - def initialize(relations:) + def initialize(readers:) super() - @relations = relations - define_readers! + include readers end # @api private @@ -30,15 +26,6 @@ def included(klass) klass.option :__registry__, default: -> { EMPTY_REGISTRY } end - - private - - # @api private - def define_readers! - relations.each do |name| - define_method(name) { __registry__[name] } - end - end end end end diff --git a/core/lib/rom/setup/finalize/finalize_relations.rb b/core/lib/rom/setup/finalize/finalize_relations.rb index e0b73656e..7b8565f64 100644 --- a/core/lib/rom/setup/finalize/finalize_relations.rb +++ b/core/lib/rom/setup/finalize/finalize_relations.rb @@ -9,6 +9,16 @@ class Finalize class FinalizeRelations attr_reader :notifications + class RegistryReaders < ::Module + def initialize(relations) + super() + + relations.each do |name| + define_method(name) { __registry__[name] } + end + end + end + # Build relation registry of specified descendant classes # # This is used by the setup @@ -32,6 +42,7 @@ def initialize(gateways, relation_classes, notifications:, mappers: nil, plugins # rubocop:disable Metrics/AbcSize, Metrics/MethodLength def run! relation_registry = RelationRegistry.new do |registry, relations| + registry_readers = RegistryReaders.new(relation_names) @relation_classes.each do |klass| unless klass.adapter raise MissingAdapterIdentifierError, @@ -45,7 +56,7 @@ def run! "Relation with name #{key.inspect} registered more than once" end - klass.use(:registry_reader, relations: relation_names) + klass.use(:registry_reader, readers: registry_readers) notifications.trigger( 'configuration.relations.class.ready', diff --git a/repository/lib/rom/repository.rb b/repository/lib/rom/repository.rb index 857ddc362..f42e56f54 100644 --- a/repository/lib/rom/repository.rb +++ b/repository/lib/rom/repository.rb @@ -80,8 +80,9 @@ class Repository # @!method self.relation_reader # Get or set relation reader module - # @return [RelationReader] + # @return [Module] defines :relation_reader + relation_reader RelationReader struct_namespace ROM::Struct diff --git a/repository/lib/rom/repository/class_interface.rb b/repository/lib/rom/repository/class_interface.rb index 1bf549348..866d5e390 100644 --- a/repository/lib/rom/repository/class_interface.rb +++ b/repository/lib/rom/repository/class_interface.rb @@ -59,9 +59,11 @@ def [](name) def new(container = nil, **options) container ||= options.fetch(:container) - unless relation_reader - relation_reader(RelationReader.new(self, container.relations.elements.keys)) - include(relation_reader) + unless self < relation_reader + include relation_reader.new( + relations: container.relations.elements.keys, + cache: container.cache + ) end super(**options, container: container) diff --git a/repository/lib/rom/repository/relation_reader.rb b/repository/lib/rom/repository/relation_reader.rb index 94359c45d..0b800f262 100644 --- a/repository/lib/rom/repository/relation_reader.rb +++ b/repository/lib/rom/repository/relation_reader.rb @@ -4,45 +4,72 @@ module ROM class Repository # @api private class RelationReader < ::Module - # @api private - attr_reader :klass - - # @api private - attr_reader :relations - module InstanceMethods + private + # @api private - def set_relation(name) # rubocop:disable Naming/AccessorMethodName + def prepare_relation(name, **) container .relations[name] - .with(auto_struct: auto_struct) - .struct_namespace(struct_namespace) + .with( + auto_struct: auto_struct, + struct_namespace: struct_namespace + ) + end + + # @api private + def relation_reader(cache, ...) + cache_key = relation_cache_key(...) + cache.fetch_or_store(*cache_key) { prepare_relation(...) } + end + + # @api private + def relation_cache_key(name, **) + [name, auto_struct, struct_namespace] end end # @api private - def initialize(klass, relations) - super() - @klass = klass - @relations = relations - define_readers! + class Readers < ::Module + # @api private + attr_reader :cache + + def initialize(relations) + super() + + include InstanceMethods + + define_readers(relations) + end + + # @api private + def define_readers(relations) + cache = Cache.new + relations.each do |name| + define_readers_for_relation(cache, name) + end + end + + # @api private + def define_readers_for_relation(cache, name) + define_method(name) do |**kwargs| + relation_reader(cache, name, **kwargs) + end + end end # @api private - def included(klass) - super - klass.include(InstanceMethods) - end + def initialize(relations:, cache:) + super() - private + add_readers(relations, cache) + end # @api private - def define_readers! - relations.each do |name| - define_method(name) do - @relations[name] ||= set_relation(name) - end - end + def add_readers(relations, cache) + include cache.fetch_or_store(:relation_readers) { + Readers.new(relations) + } end end end diff --git a/repository/lib/rom/repository/root.rb b/repository/lib/rom/repository/root.rb index be0c891be..17d0a57dc 100644 --- a/repository/lib/rom/repository/root.rb +++ b/repository/lib/rom/repository/root.rb @@ -58,7 +58,7 @@ def self.inherited(klass) # @see Repository#initialize def initialize(*, **) super - @root = set_relation(self.class.root) + @root = prepare_relation(self.class.root) end end end diff --git a/repository/spec/integration/plugin_spec.rb b/repository/spec/integration/plugin_spec.rb index 641b129be..a500e5f24 100644 --- a/repository/spec/integration/plugin_spec.rb +++ b/repository/spec/integration/plugin_spec.rb @@ -12,7 +12,7 @@ def self.apply(target, **) target.prepend(self) end - def set_relation(*) + def prepare_relation(*) super.where { `1 = 0` } end end