mirror of
https://github.com/ash-project/ash_postgres.git
synced 2024-09-20 21:43:12 +12:00
36 lines
873 B
Elixir
36 lines
873 B
Elixir
|
defmodule AshPostgres.Test.Concat do
|
||
|
@moduledoc false
|
||
|
use Ash.Calculation
|
||
|
require Ash.Query
|
||
|
|
||
|
def init(opts) do
|
||
|
if opts[:keys] && is_list(opts[:keys]) && Enum.all?(opts[:keys], &is_atom/1) do
|
||
|
{:ok, opts}
|
||
|
else
|
||
|
{:error, "Expected a `keys` option for which keys to concat"}
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def expression(opts, %{separator: separator}) do
|
||
|
Enum.reduce(opts[:keys], nil, fn key, expr ->
|
||
|
if expr do
|
||
|
if separator do
|
||
|
Ash.Query.expr(^expr <> ^separator <> ref(^key))
|
||
|
else
|
||
|
Ash.Query.expr(^expr <> ref(^key))
|
||
|
end
|
||
|
else
|
||
|
Ash.Query.expr(ref(^key))
|
||
|
end
|
||
|
end)
|
||
|
end
|
||
|
|
||
|
def calculate(records, opts, %{separator: separator}) do
|
||
|
Enum.map(records, fn record ->
|
||
|
Enum.map_join(opts[:keys], separator, fn key ->
|
||
|
to_string(Map.get(record, key))
|
||
|
end)
|
||
|
end)
|
||
|
end
|
||
|
end
|