Skip to content

Commit

Permalink
Add support for remaining *_any/*_all attribute methods, and add matc…
Browse files Browse the repository at this point in the history
…hes/does_not_match/not_in
  • Loading branch information
Ernie Miller committed Sep 29, 2010
1 parent 261d284 commit 3a994b9
Show file tree
Hide file tree
Showing 7 changed files with 597 additions and 5 deletions.
133 changes: 129 additions & 4 deletions lib/arel/attributes/attribute.rb
Original file line number Diff line number Diff line change
Expand Up @@ -7,16 +7,24 @@ def not_eq other
Nodes::NotEqual.new self, other
end

def not_eq_any others
grouping_any :not_eq, others
end

def not_eq_all others
grouping_all :not_eq, others
end

def eq other
Nodes::Equality.new self, other
end

def eq_any others
first = Nodes::Equality.new self, others.shift
grouping_any :eq, others
end

Nodes::Grouping.new others.inject(first) { |memo,expr|
Nodes::Or.new(memo, Nodes::Equality.new(self, expr))
}
def eq_all others
grouping_all :eq, others
end

def in other
Expand All @@ -36,13 +44,130 @@ def in other
end
end

def in_any others
grouping_any :in, others
end

def in_all others
grouping_all :in, others
end

def not_in other
case other
when Arel::SelectManager
Nodes::NotIn.new self, other.to_a.map { |x| x.id }
when Range
if other.exclude_end?
left = Nodes::LessThan.new(self, other.min)
right = Nodes::GreaterThanOrEqual.new(self, other.max)
Nodes::Or.new left, right
else
left = Nodes::LessThan.new(self, other.min)
right = Nodes::GreaterThan.new(self, other.max)
Nodes::Or.new left, right
end
else
Nodes::NotIn.new self, other
end
end

def not_in_any others
grouping_any :not_in, others
end

def not_in_all others
grouping_all :not_in, others
end

def matches other
Nodes::Matches.new self, other
end

def matches_any others
grouping_any :matches, others
end

def matches_all others
grouping_all :matches, others
end

def does_not_match other
Nodes::DoesNotMatch.new self, other
end

def does_not_match_any others
grouping_any :does_not_match, others
end

def does_not_match_all others
grouping_all :does_not_match, others
end

def gteq right
Nodes::GreaterThanOrEqual.new self, right
end

def gteq_any others
grouping_any :gteq, others
end

def gteq_all others
grouping_all :gteq, others
end

def gt right
Nodes::GreaterThan.new self, right
end

def gt_any others
grouping_any :gt, others
end

def gt_all others
grouping_all :gt, others
end

def lt right
Nodes::LessThan.new self, right
end

def lt_any others
grouping_any :lt, others
end

def lt_all others
grouping_all :lt, others
end

def lteq right
Nodes::LessThanOrEqual.new self, right
end

def lteq_any others
grouping_any :lteq, others
end

def lteq_all others
grouping_all :lteq, others
end

private

def grouping_any method_id, others
first = send method_id, others.shift

Nodes::Grouping.new others.inject(first) { |memo,expr|
Nodes::Or.new(memo, send(method_id, expr))
}
end

def grouping_all method_id, others
first = send method_id, others.shift

Nodes::Grouping.new others.inject(first) { |memo,expr|
Nodes::And.new(memo, send(method_id, expr))
}
end
end

class String < Attribute; end
Expand Down
3 changes: 3 additions & 0 deletions lib/arel/nodes.rb
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,11 @@
require 'arel/nodes/greater_than_or_equal'
require 'arel/nodes/less_than'
require 'arel/nodes/less_than_or_equal'
require 'arel/nodes/matches'
require 'arel/nodes/does_not_match'

require 'arel/nodes/in'
require 'arel/nodes/not_in'
require 'arel/nodes/lock'
require 'arel/nodes/function'
require 'arel/nodes/count'
Expand Down
6 changes: 6 additions & 0 deletions lib/arel/nodes/does_not_match.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
module Arel
module Nodes
class DoesNotMatch < Binary
end
end
end
6 changes: 6 additions & 0 deletions lib/arel/nodes/matches.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
module Arel
module Nodes
class Matches < Binary
end
end
end
6 changes: 6 additions & 0 deletions lib/arel/nodes/not_in.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
module Arel
module Nodes
class NotIn < Binary
end
end
end
14 changes: 14 additions & 0 deletions lib/arel/visitors/to_sql.rb
Original file line number Diff line number Diff line change
Expand Up @@ -161,6 +161,14 @@ def visit_Arel_Nodes_LessThan o
"#{visit o.left} < #{visit o.right}"
end

def visit_Arel_Nodes_Matches o
"#{visit o.left} LIKE #{visit o.right}"
end

def visit_Arel_Nodes_DoesNotMatch o
"#{visit o.left} NOT LIKE #{visit o.right}"
end

def visit_Arel_Nodes_StringJoin o
"#{visit o.left} #{visit o.right}"
end
Expand Down Expand Up @@ -191,6 +199,12 @@ def visit_Arel_Nodes_In o
"#{visit o.left} IN (#{right})"
end

def visit_Arel_Nodes_NotIn o
right = o.right
right = right.empty? ? 'NULL' : right.map { |x| visit x }.join(', ')
"#{visit o.left} NOT IN (#{right})"
end

def visit_Arel_Nodes_And o
"#{visit o.left} AND #{visit o.right}"
end
Expand Down
Loading

0 comments on commit 3a994b9

Please sign in to comment.