Class: Wallaby::Map::ModelClassMapper

Inherits:
Object
  • Object
show all
Defined in:
lib/services/wallaby/map/model_class_mapper.rb

Overview

To map model class to a klass

Instance Method Summary collapse

Constructor Details

#initialize(base_class) ⇒ ModelClassMapper 



5
6
7
 
# File 'lib/services/wallaby/map/model_class_mapper.rb', line 5

def initialize(base_class)
  @base_class = base_class
end

Instance Method Details

#map ⇒ Object 



9
10
11
12
13
14
15
 
# File 'lib/services/wallaby/map/model_class_mapper.rb', line 9

def map
  (@base_class.try(:subclasses) || EMPTY_HASH)
    .each_with_object({}) do |klass, map|
    next if anonymous? klass
    map[klass.model_class] = block_given? ? yield(klass) : klass
  end
end