Class: Wallaby::Map::ModeMapper

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

Overview

To map model class to mode class so that we know how to handle a model

Instance Method Summary collapse

Constructor Details

#initialize(mode_classes) ⇒ ModeMapper

Returns a new instance of ModeMapper.



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

def initialize(mode_classes)
  @mode_classes = mode_classes
end

Instance Method Details

#map ⇒ Object 



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

def map
  return {} if @mode_classes.blank?
  @mode_classes.each_with_object({}) do |mode_class, map|
    mode_class.model_finder.new.all.each do |model_class|
      map[model_class] = mode_class
    end
  end
end