mirror of
https://github.com/ash-project/ash_postgres.git
synced 2024-09-21 05:53:18 +12:00
37cc01957d
* WIP * chore: fix mix.lock merge issues * improvement: upgrade to 3.0 * chore: remove `repo.to_tenant` * chore: continue removal of unnecessary helper * chore: use `Ash.ToTenant`
35 lines
866 B
Elixir
35 lines
866 B
Elixir
defmodule AshPostgres.Test.Concat do
|
|
@moduledoc false
|
|
use Ash.Resource.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, %{arguments: %{separator: separator}}) do
|
|
Enum.reduce(opts[:keys], nil, fn key, expr ->
|
|
if expr do
|
|
if separator do
|
|
expr(^expr <> ^separator <> ^ref(key))
|
|
else
|
|
expr(^expr <> ^ref(key))
|
|
end
|
|
else
|
|
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
|