FlyingNoodle

Return a struct of a given type from a calculation?

Given this code:

relationships do
    belongs_to :sender, MyApp.Accounts.User,
      source_attribute: :sender_id,
      destination_attribute: :id,
      allow_nil?: false,
      attribute_writable?: true

    belongs_to :recipient, MyApp.Accounts.User,
      source_attribute: :recipient_id,
      destination_attribute: :id,
      allow_nil?: false,
      attribute_writable?: true

    has_many :messages, MyApp.Conversations.Message
  end

  calculations do
    calculate :other_user,
              :struct,
              expr(
                if sender_id == ^arg(:current_user_id) do
                  recipient
                else
                  sender
                end
              ),
              constraints: [instance_of: MyApp.Accounts.User] do
      argument :current_user_id, :uuid, allow_nil?: false
    end
  end

I would like the calculation other_user to return an instance of the MyApp.Accounts.User resource. However, I get the error that: error: undefined function calculate/5 (there is no such import).

2 comments

/ash #ash-calculations

1 343 2

2023-12-14 14:58:57 UTC

Marked As Solved

zachdaniel

Creator of Ash

So, you won’t be able to this using an expression calculation, you’ll need to use an Elixir calculation. This is because expression calculations done in the data layer can’t return related resources, only relationships and runtime calculations can do that.

defmodule RecipientOrSender do
  use Ash.Calculation

  def load(_, _, _), do: [:recipient, :sender]
  
  def calculate(records, _, context) do
     Enum.map(records, fn record -> 
       if record.recipient.id == context.current_user_id do
         record.recipient
       else
         record.sender
       end
     end)
  end
end

Then you can do:

    calculate :other_user,
              :struct,
              expr(
                if sender_id == ^arg(:current_user_id) do
                  recipient
                else
                  sender
                end
              ) do
      constraints [instance_of: MyApp.Accounts.User]
      argument :current_user_id, :uuid, allow_nil?: false
    end

Although, come to think of it, you could use a manual relationship, assuming you’re okay with using the actor in the standard way as opposed to passing in a current_user_id.

defmodule OtherUser do
  use Ash.Resource.ManualRelationship

  def load(records, _, %{api: api, actor: %{id: actor_id}}) do
    records
    |> Api.load!([:sender, :recipient])
    |> Map.new(fn record -> 
      {record.id, [record.sender, record.recipient] |> Enum.filter(&(&1.id == actor_id))}
    end)
  end

  def load(_, _, _), do: []
end

Haven’t actually run the above code snippets, they are just examples to get you going.

Post #2

Also Liked

FlyingNoodle

zachdaniel:

defmodule OtherUser do
  use Ash.Resource.ManualRelationship

  def load(records, _, %{api: api, actor: %{id: actor_id}}) do
    records
    |> Api.load!([:sender, :recipient])
    |> Map.new(fn record -> 
      {record.id, [record.sender, record.recipient] |> Enum.filter(&(&1.id == actor_id))}
    end)
  end

  def load(_, _, _), do: []
end

This was almost right, This is correct:

def load(records, _, %{api: api, actor: %{id: actor_id}}) do
    {:ok, records
    |> MyApi.load!([:sender, :recipient])
    |> Map.new(fn record -> 
      {record.id, [record.sender, record.recipient] |> Enum.filter(&(&1.id == actor_id))}
    end)}
  end

I was most worried about this making one query per result but it actually doesn’t and works very nicely!

[debug] QUERY OK source="users" db=0.3ms idle=1855.7ms
SELECT u0."id", u0."email", u0."hashed_password" FROM "users" AS u0 LIMIT $1 [1]
↳ AshPostgres.DataLayer.run_query/2, at: lib/data_layer.ex:648
[debug] QUERY OK source="conversations" db=0.5ms idle=1856.8ms
SELECT c0."id", c0."sender_id", c0."recipient_id" FROM "conversations" AS c0 WHERE ((c0."sender_id"::uuid = $1::uuid) OR (c0."recipient_id"::uuid = $2::uuid)) ["d065715d-14fb-4eeb-9e4c-867b41e7ea18", "d065715d-14fb-4eeb-9e4c-867b41e7ea18"]
↳ AshPostgres.DataLayer.run_query/2, at: lib/data_layer.ex:648
[debug] QUERY OK source="users" db=1.2ms idle=1859.4ms
SELECT u0."id", u0."email", u0."hashed_password" FROM "users" AS u0 WHERE (u0."id"::uuid IN ($1::uuid,$2::uuid,$3::uuid,$4::uuid)) ["80cef98a-ed05-4955-9fce-114b54467d4c", "834e6d70-7a48-4065-ae1c-bc8f6c8fbc0e", "99c67645-2208-4185-9961-7517d4779222", "d065715d-14fb-4eeb-9e4c-867b41e7ea18"]
↳ AshPostgres.DataLayer.run_query/2, at: lib/data_layer.ex:648
[debug] QUERY OK source="users" db=0.4ms queue=0.3ms idle=1860.1ms
SELECT u0."id", u0."email", u0."hashed_password" FROM "users" AS u0 WHERE (u0."id"::uuid IN ($1::uuid,$2::uuid,$3::uuid,$4::uuid,$5::uuid,$6::uuid,$7::uuid,$8::uuid,$9::uuid,$10::uuid,$11::uuid,$12::uuid,$13::uuid,$14::uuid,$15::uuid,$16::uuid,$17::uuid,$18::uuid,$19::uuid,$20::uuid,$21::uuid,$22::uuid,$23::uuid,$24::uuid,$25::uuid,$26::uuid,$27::uuid,$28::uuid,$29::uuid,$30::uuid,$31::uuid,$32::uuid,$33::uuid,$34::uuid,$35::uuid,$36::uuid,$37::uuid,$38::uuid,$39::uuid,$40::uuid,$41::uuid,$42::uuid,$43::uuid,$44::uuid,$45::uuid,$46::uuid,$47::uuid,$48::uuid,$49::uuid,$50::uuid,$51::uuid,$52::uuid,$53::uuid,$54::uuid,$55::uuid,$56::uuid,$57::uuid,$58::uuid,$59::uuid,$60::uuid,$61::uuid,$62::uuid,$63::uuid)) ["4e42a0a2-f982-422a-bef3-f77bc3fab5ed", "fbee0b56-50a6-45fb-b9e0-b48103f82f61", "2c97d01a-c00b-4186-9ac2-70f649cad75b", "5150363b-d8cd-4f41-9c81-06b3bcc5faae", "6cfbfa16-4283-478b-bb7e-9d39ddc02c4e", "ce9afdd2-6575-4422-ac4d-b3ffe34aab5c", "941a544a-0c7c-4a72-a531-261706bd6b41", "f8b927a5-27b4-4258-8fef-87b06aec40af", "80cef98a-ed05-4955-9fce-114b54467d4c", "2febf7fc-2e7e-4d10-8e8e-42980bfee682", "4cee3e89-2c4b-4641-9a68-43b4cfc3f629", "9370c439-9ef0-40a3-a545-d10ff68bfa14", "9ed08237-e33f-4282-a9b4-b8c581a0c46e", "bd8b24c9-2202-44fc-9e1e-4b3e92acdbb4", "1412e573-5106-4bcb-aa05-6edbfb9c1040", "28fac61f-bdd7-4a52-bbb1-fc8596056648", "f57bdfe1-aeee-41c1-8ac6-e0af2a0bb848", "49eee110-1c95-4c59-bbda-6904b66b5acb", "cbbae368-03a4-4b3b-a5dc-be6f0eebd0f8", "dc80d246-57f1-43d0-abd5-5138678acfa7", "81bb4fe1-90b8-4a88-9473-0bff35a127c9", "d2b8ffd8-0c95-4aad-9786-a5ac10f07c40", "5f3597f4-740f-41c2-a307-b0353f195c7d", "a9169382-ad06-4477-8b6c-ac4b60c43003", "2d4840dd-2d18-4158-838a-47e3060a9e13", "04ebfc1c-56b0-48a7-92b6-faa02a7cefeb", "25efce68-88a7-4f41-a161-73a0e6f9d4ef", "544fec19-ef81-4dab-8c27-187d3df0a04a", "834e6d70-7a48-4065-ae1c-bc8f6c8fbc0e", "b9567c52-c635-410d-abc0-2f8a3b7d07a2", "50272c49-0640-491e-8645-2dcdacda5f53", "db816428-8204-4270-b077-fa7213e85514", "82d054fc-5a82-4abc-9f21-e5c62d8461da", "99c67645-2208-4185-9961-7517d4779222", "b073f1a9-bc07-4f83-81d4-520b036ad651", "b3d32c08-82c9-4fc8-a7f2-d46d89bd211a", "064d3055-45b2-4f92-b3f5-0ddeb24382a6", "53415340-08d7-4945-bd54-344a45f29a41", "3f258315-194a-4b2e-a031-9d5638e0fe4c", "b4ee2c5a-06b3-45b0-b0f1-31c94a1dae4c", "de893910-4994-438a-a02b-2db7bee5cdca", "ca120785-7c9b-43a4-974a-a98e5d0a9084", "77de8899-43c3-43d2-a2ff-dcb97439c7f8", "85309f25-a09b-4182-8152-ae6cde0a41ba", "c39f04d5-6107-4f3e-a637-2661d62910e7", "d065715d-14fb-4eeb-9e4c-867b41e7ea18", "103c790d-5d2d-4888-8a41-800cd97bf6ed", "e97c3bdb-5f6b-44a0-a1f9-a4be3f0b29cf", "7d6f7f58-ab20-4309-a7f0-9d39f76c6663", "913e38d9-562e-4a49-8089-3b5d0602be28", ...]

Post #3

Where Next?

View thread on forum (has 2 responses!)

ash

ash-calculations

Home Questions & Help>Questions

/ash #ash-calculations

1 348 2

Last post

Return a struct of a given type from a calculation?

FlyingNoodle

Return a struct of a given type from a calculation?

Marked As Solved

zachdaniel

Also Liked

FlyingNoodle

Where Next?

Popular in Questions

How to set environment variables in dev.exs?

Typespecs - best way to spec keyword lists?

How to convert Date to DateTime which Ecto needs

Understanding Elixir/Phoenix performance

How to decode a JSON into a struct safely?

BEAM file compliation error after degrading Erlang version from 22 to 21

Installing elixir via asdf shows zsh: command not found: iex

Removing spaces and characters from string

Form submit on Enter keypress for textarea input type

How To Implement if...else if...else condition

Other popular topics

Oban - Reliable and Observable Job Processing

Which Neovim configuration distro do you use/recommend? (Poll)

Best way to send multiple files as HTTP response

(Postgrex.Error) FATAL 28P01 (invalid_password) password authentication failed for user “postgres”

How can I check Phoenix version?

Phoenix 1.4.0 released!

Surface - A component-based library for Phoenix LiveView

How do I kill a process ` #PID<0.186.0` in iex?

Installing elixir via asdf shows zsh: command not found: iex

Concat/appending lists

Latest Ash Threads

Questions & Help>Questions

Latest on Elixir Forum

Sponsor Spotlight

Our Sponsors

Categories:

Sub Categories:

Forums

Popular Tags

Our Sponsors

We're in Beta