diff options
author | Rafael Mendonça França <rafaelmfranca@gmail.com> | 2013-07-29 19:32:28 -0700 |
---|---|---|
committer | Rafael Mendonça França <rafaelmfranca@gmail.com> | 2013-07-29 19:32:28 -0700 |
commit | 43187f9cdc631f9ee8ec68025d81113acdf8eeef (patch) | |
tree | 16b0df9d1dbe7cf100f0a886855181ea16bf2c56 /activerecord/lib/active_record/relation/predicate_builder/array_handler.rb | |
parent | fbaae891ac5361a03aaea07bc95bc11b3c6e11fd (diff) | |
parent | 92a603387c084f13a36bbf3844d89029bb73a753 (diff) | |
download | rails-43187f9cdc631f9ee8ec68025d81113acdf8eeef.tar.gz rails-43187f9cdc631f9ee8ec68025d81113acdf8eeef.tar.bz2 rails-43187f9cdc631f9ee8ec68025d81113acdf8eeef.zip |
Merge pull request #10673 from sgrif/master
Add ability to specify how a class is converted to Arel predicate when passed to where
Diffstat (limited to 'activerecord/lib/active_record/relation/predicate_builder/array_handler.rb')
-rw-r--r-- | activerecord/lib/active_record/relation/predicate_builder/array_handler.rb | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/activerecord/lib/active_record/relation/predicate_builder/array_handler.rb b/activerecord/lib/active_record/relation/predicate_builder/array_handler.rb new file mode 100644 index 0000000000..2f6c34ac08 --- /dev/null +++ b/activerecord/lib/active_record/relation/predicate_builder/array_handler.rb @@ -0,0 +1,29 @@ +module ActiveRecord + class PredicateBuilder + class ArrayHandler # :nodoc: + def call(attribute, value) + values = value.map { |x| x.is_a?(Base) ? x.id : x } + ranges, values = values.partition { |v| v.is_a?(Range) } + + values_predicate = if values.include?(nil) + values = values.compact + + case values.length + when 0 + attribute.eq(nil) + when 1 + attribute.eq(values.first).or(attribute.eq(nil)) + else + attribute.in(values).or(attribute.eq(nil)) + end + else + attribute.in(values) + end + + array_predicates = ranges.map { |range| attribute.in(range) } + array_predicates << values_predicate + array_predicates.inject { |composite, predicate| composite.or(predicate) } + end + end + end +end |