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.



4
5
6
# File 'lib/active_record/relation/predicate_builder/array_handler.rb', line 4

def initialize(predicate_builder)
  @predicate_builder = predicate_builder
end

Instance Method Details

#call(attribute, value) ⇒ Object



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# File 'lib/active_record/relation/predicate_builder/array_handler.rb', line 8

def call(attribute, value)
  values = value.map { |x| x.is_a?(Base) ? x.id : x }
  nils, values = values.partition(&:nil?)

  return attribute.in([]) if values.empty? && nils.empty?

  ranges, values = values.partition { |v| v.is_a?(Range) }

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

  unless nils.empty?
    values_predicate = values_predicate.or(predicate_builder.build(attribute, nil))
  end

  array_predicates = ranges.map { |range| predicate_builder.build(attribute, range) }
  array_predicates.unshift(values_predicate)
  array_predicates.inject { |composite, predicate| composite.or(predicate) }
end