aboutsummaryrefslogtreecommitdiffstats
path: root/lib/arel/predicates.rb
blob: b639022b4e1f39e4e91a0ff5f8ff8c17befab43b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
module Arel
  class Predicate
    def or(other_predicate)
      Or.new(self, other_predicate)
    end

    def and(other_predicate)
      And.new(self, other_predicate)
    end
  end

  class Binary < Predicate
    attributes :operand1, :operand2
    deriving :initialize

    def ==(other)
      self.class === other          and
      @operand1  ==  other.operand1 and
      @operand2  ==  other.operand2
    end

    def bind(relation)
      self.class.new(operand1.find_correlate_in(relation), operand2.find_correlate_in(relation))
    end

    def to_sql(formatter = nil)
      "#{operand1.to_sql} #{predicate_sql} #{operand1.format(operand2)}"
    end
    alias_method :to_s, :to_sql
  end

  class CompoundPredicate < Binary
    def to_sql(formatter = nil)
      "(#{operand1.to_sql(formatter)} #{predicate_sql} #{operand2.to_sql(formatter)})"
    end
  end

  class Or < CompoundPredicate
    def predicate_sql; "OR" end
  end

  class And < CompoundPredicate
    def predicate_sql; "AND" end
  end

  class Equality < Binary
    def ==(other)
      Equality === other and
        ((operand1 == other.operand1 and operand2 == other.operand2) or
         (operand1 == other.operand2 and operand2 == other.operand1))
    end

    def predicate_sql
      operand2.equality_predicate_sql
    end
  end

  class GreaterThanOrEqualTo < Binary
    def predicate_sql; '>=' end
  end

  class GreaterThan < Binary
    def predicate_sql; '>' end
  end

  class LessThanOrEqualTo < Binary
    def predicate_sql; '<=' end
  end

  class LessThan < Binary
    def predicate_sql; '<' end
  end

  class Match < Binary
    def predicate_sql; 'LIKE' end
  end

  class In < Binary
    def predicate_sql; operand2.inclusion_predicate_sql end
  end
end