Class: Wallaby::Map::ModelClassCollector

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

Overview

To collect model classes that are configured to be handled by Wallaby

Instance Method Summary collapse

Constructor Details

#initialize(configuration) ⇒ ModelClassCollector

Returns a new instance of ModelClassCollector.

Parameters:



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

def initialize(configuration)
  @configuration = configuration
end

Instance Method Details

#collect ⇒ Array<Class>

Returns model class.

Returns:

  • (Array<Class>) —

    model class



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

def collect
  return all_models - excluded_models if configured_models.blank?
  invalid_models_check
  configured_models
end