Class: TypeBalancer::TypeExtractor

Inherits:

Object

• Object
• TypeBalancer::TypeExtractor

Defined in:

lib/type_balancer/type_extractor.rb

Instance Method Summary

• #extract_types(collection) ⇒ Object
• #group_by_type(collection) ⇒ Object
• #initialize(type_field) ⇒ TypeExtractor constructor

  A new instance of TypeExtractor.

Constructor Details

#initialize(type_field) ⇒ TypeExtractor

Returns a new instance of TypeExtractor.

# File 'lib/type_balancer/type_extractor.rb', line 5

def initialize(type_field)
  @type_field = type_field
end

Instance Method Details

#extract_types(collection) ⇒ Object

# File 'lib/type_balancer/type_extractor.rb', line 9

def extract_types(collection)
  collection.map { |item| get_type(item) }.uniq
end

#group_by_type(collection) ⇒ Object

# File 'lib/type_balancer/type_extractor.rb', line 13

def group_by_type(collection)
  collection.group_by { |item| get_type(item) }
end