Skip to content

Commit

Permalink
improvement: more clean boolean filter optimization for or ==
Browse files Browse the repository at this point in the history
  • Loading branch information
zachdaniel committed Feb 5, 2025
1 parent de02fbb commit f2cb11c
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 4 deletions.
18 changes: 14 additions & 4 deletions lib/ash/query/boolean_expression.ex
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,14 @@ defmodule Ash.Query.BooleanExpression do
%__MODULE__{op: op, left: left, right: right}
end

defp maybe_optimize(%__MODULE__{op: op, left: left, right: right}) do
optimized_new(op, left, right)
end

defp maybe_optimize(other) do
other
end

# In many cases we could actually just return `true/false` directly because we know
# statements are contradictions. However, that would likely confuse users. For example:
# `Ash.Query.filter(Resource, x == 1 and x in [2, 3])` becomes `#Ash.Query<filter: false>`
Expand All @@ -29,12 +37,14 @@ defmodule Ash.Query.BooleanExpression do
def optimized_new(:and, _, false), do: false
def optimized_new(:and, nil, _), do: false
def optimized_new(:and, _, nil), do: false
def optimized_new(:and, true, right), do: right
def optimized_new(:and, left, true), do: left
def optimized_new(:and, true, right), do: maybe_optimize(right)
def optimized_new(:and, left, true), do: maybe_optimize(left)
def optimized_new(:or, true, _), do: true
def optimized_new(:or, _, true), do: true
def optimized_new(:or, nil, right), do: right
def optimized_new(:or, left, nil), do: left
def optimized_new(:or, nil, right), do: maybe_optimize(right)
def optimized_new(:or, false, right), do: maybe_optimize(right)
def optimized_new(:or, left, nil), do: maybe_optimize(left)
def optimized_new(:or, left, false), do: maybe_optimize(left)

def optimized_new(
op,
Expand Down
27 changes: 27 additions & 0 deletions test/filter/filter_test.exs
Original file line number Diff line number Diff line change
Expand Up @@ -307,6 +307,33 @@ defmodule Ash.Test.Filter.FilterTest do
assert stringified_query =~ ~S(title in ["bar", "baz", "foo"])
end

test "multiple nested equality simplifies to `in`" do
import Ash.Expr
keys = [:id]

records =
for i <- ["foo", "bar", "baz", "buz"] do
%{id: i}
end

expr =
Enum.reduce(records, expr(false), fn record, filter_expr ->
all_keys_match_expr =
Enum.reduce(keys, expr(true), fn key, key_expr ->
expr(^key_expr and ^ref(key) == ^Map.get(record, key))
end)

expr(^filter_expr or ^all_keys_match_expr)
end)

stringified_query =
Post
|> Ash.Query.filter(^expr)
|> inspect()

assert stringified_query =~ ~S(id in ["bar", "baz", "buz", "foo"])
end

test "in across ands in ors isn't optimized" do
stringified_query =
Post
Expand Down

0 comments on commit f2cb11c

Please sign in to comment.