forked from brynary/arel
-
Notifications
You must be signed in to change notification settings - Fork 390
/
predications.rb
188 lines (156 loc) · 4.83 KB
/
predications.rb
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
module Arel
module Predications
def not_eq other
Nodes::NotEqual.new self, Nodes.build_quoted(other, self)
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, Nodes.build_quoted(other, self)
end
def eq_any others
grouping_any :eq, others
end
def eq_all others
grouping_all :eq, others.map { |x| Nodes.build_quoted(x, self) }
end
def in other
case other
when Arel::SelectManager
Arel::Nodes::In.new(self, other.ast)
when Range
if other.begin == -Float::INFINITY
if other.end == Float::INFINITY
Nodes::NotIn.new self, []
elsif other.exclude_end?
Nodes::LessThan.new(self, Nodes.build_quoted(other.end, self))
else
Nodes::LessThanOrEqual.new(self, Nodes.build_quoted(other.end, self))
end
elsif other.end == Float::INFINITY
Nodes::GreaterThanOrEqual.new(self, Nodes.build_quoted(other.begin, self))
elsif other.exclude_end?
left = Nodes::GreaterThanOrEqual.new(self, Nodes.build_quoted(other.begin, self))
right = Nodes::LessThan.new(self, Nodes.build_quoted(other.end, self))
Nodes::And.new [left, right]
else
Nodes::Between.new(self, Nodes::And.new([Nodes.build_quoted(other.begin, self), Nodes.build_quoted(other.end, self)]))
end
when Array
Nodes::In.new self, other.map { |x| Nodes.build_quoted(x, self) }
else
Nodes::In.new self, Nodes.build_quoted(other, self)
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
Arel::Nodes::NotIn.new(self, other.ast)
when Range
if other.begin == -Float::INFINITY # The range begins with negative infinity
if other.end == Float::INFINITY
Nodes::In.new self, [] # The range is infinite, so return an empty range
elsif other.exclude_end?
Nodes::GreaterThanOrEqual.new(self, Nodes.build_quoted(other.end, self))
else
Nodes::GreaterThan.new(self, Nodes.build_quoted(other.end, self))
end
elsif other.end == Float::INFINITY
Nodes::LessThan.new(self, Nodes.build_quoted(other.begin, self))
else
left = Nodes::LessThan.new(self, Nodes.build_quoted(other.begin, self))
if other.exclude_end?
right = Nodes::GreaterThanOrEqual.new(self, Nodes.build_quoted(other.end, self))
else
right = Nodes::GreaterThan.new(self, Nodes.build_quoted(other.end, self))
end
Nodes::Or.new left, right
end
when Array
Nodes::NotIn.new self, other.map { |x| Nodes.build_quoted(x, self) }
else
Nodes::NotIn.new self, Nodes.build_quoted(other, self)
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, Nodes.build_quoted(other, self)
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, Nodes.build_quoted(other, self)
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, Nodes.build_quoted(right, self)
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, Nodes.build_quoted(right, self)
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
nodes = others.map {|expr| send(method_id, expr)}
Nodes::Grouping.new nodes.inject { |memo,node|
Nodes::Or.new(memo, node)
}
end
def grouping_all method_id, others
Nodes::Grouping.new Nodes::And.new(others.map {|expr| send(method_id, expr)})
end
end
end