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.

Parameters:

  • model_class (Class) —

    model class



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

def initialize(mode_classes)
  @mode_classes = mode_classes
end

Instance Method Details

#map ⇒ Hash

Returns { model_class => mode }.

Returns:

  • (Hash) —

    { model_class => mode }



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

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