Class: ActiveRecord::PredicateBuilder::ArrayHandler

Inherits:
Object
  • Object
show all
Defined in:
lib/active_record/relation/predicate_builder/array_handler.rb

Overview

:nodoc:

Defined Under Namespace

Modules: NullPredicate

Instance Method Summary collapse

Constructor Details

#initialize(predicate_builder) ⇒ ArrayHandler

Returns a new instance of ArrayHandler.



8
9
10
# File 'lib/active_record/relation/predicate_builder/array_handler.rb', line 8

def initialize(predicate_builder)
  @predicate_builder = predicate_builder
end

Instance Method Details

#call(attribute, value) ⇒ Object



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# File 'lib/active_record/relation/predicate_builder/array_handler.rb', line 12

def call(attribute, value)
  return attribute.in([]) if value.empty?

  values = value.map { |x| x.is_a?(Base) ? x.id : x }
  nils = values.extract!(&:nil?)
  ranges = values.extract! { |v| v.is_a?(Range) }

  values_predicate =
    case values.length
    when 0 then NullPredicate
    when 1 then predicate_builder.build(attribute, values.first)
    else Arel::Nodes::HomogeneousIn.new(values, attribute, :in)
    end

  unless nils.empty?
    values_predicate = values_predicate.or(attribute.eq(nil))
  end

  if ranges.empty?
    values_predicate
  else
    array_predicates = ranges.map! { |range| predicate_builder.build(attribute, range) }
    array_predicates.inject(values_predicate, &:or)
  end
end