2021-06-04 17:48:35 +12:00
|
|
|
defmodule AshPostgres.Test.Concat do
|
|
|
|
@moduledoc false
|
2024-03-28 09:52:28 +13:00
|
|
|
use Ash.Resource.Calculation
|
2021-06-04 17:48:35 +12:00
|
|
|
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
|
|
|
|
|
2024-03-28 09:52:28 +13:00
|
|
|
def expression(opts, %{arguments: %{separator: separator}}) do
|
2021-06-04 17:48:35 +12:00
|
|
|
Enum.reduce(opts[:keys], nil, fn key, expr ->
|
|
|
|
if expr do
|
|
|
|
if separator do
|
2024-03-28 09:52:28 +13:00
|
|
|
expr(^expr <> ^separator <> ^ref(key))
|
2021-06-04 17:48:35 +12:00
|
|
|
else
|
2024-03-28 09:52:28 +13:00
|
|
|
expr(^expr <> ^ref(key))
|
2021-06-04 17:48:35 +12:00
|
|
|
end
|
|
|
|
else
|
2024-03-28 09:52:28 +13:00
|
|
|
expr(^ref(key))
|
2021-06-04 17:48:35 +12:00
|
|
|
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
|