提交 110e0e1f 编写于 作者: M Maxime Lapointe

Avoid duplicate clauses when using #or

Condenses the clauses that are common to both sides of the OR and put them outside, before the OR
This fix the current behavior where the number of conditions is exponential based on the number of times #or is used.
上级 6f9b01c0
* When using #or, extract the common conditions and put them before the OR condition.
*Maxime Handfield Lapointe*
* `Relation#or` now accepts two relations who have different values for
`references` only, as `references` can be implicitly called by `where`.
......
......@@ -15,6 +15,12 @@ def +(other)
)
end
def -(other)
WhereClause.new(
predicates - other.predicates,
)
end
def merge(other)
WhereClause.new(
predicates_unreferenced_by(other) + other.predicates,
......@@ -26,15 +32,16 @@ def except(*columns)
end
def or(other)
if empty?
self
elsif other.empty?
other
else
WhereClause.new(
[ast.or(other.ast)],
)
end
left = self - other
common = self - left
right = other - common
return common if left.empty? || right.empty?
or_clause = WhereClause.new(
[left.ast.or(right.ast)]
)
common + or_clause
end
def to_h(table_name = nil)
......
......@@ -181,6 +181,39 @@ class WhereClauseTest < ActiveRecord::TestCase
assert_equal WhereClause.empty, WhereClause.empty.or(where_clause)
end
test "or places common conditions before the OR" do
wcs = (0..6).map do |i|
WhereClause.new([table["col_#{i}"].eq(bind_param(i))])
end
actual = wcs[0]
expected = wcs[0]
actual = (actual + wcs[1]).or(actual + wcs[2])
expected = expected + wcs[1].or(wcs[2])
actual = (actual + wcs[3] + wcs[4]).or(actual + wcs[5] + wcs[6])
expected = expected + (wcs[3] + wcs[4]).or(wcs[5] + wcs[6])
assert_equal expected, actual
end
test "or will use common conditions only if one side only has common conditions" do
common = WhereClause.new(
[table["id"].eq(bind_param(1)), "foo = bar"]
)
extra = WhereClause.new([table["extra"].eq(bind_param("pluto"))])
actual_extra_only_on_left = (common + extra).or(common)
actual_extra_only_on_right = (common).or(common + extra)
expected = common
assert_equal expected, actual_extra_only_on_left
assert_equal expected, actual_extra_only_on_right
end
private
def table
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册