ash_graphql/test/support/resources/channel/channel.ex

defmodule AshGraphql.Test.Channel do
  @moduledoc false

  use Ash.Resource,
    data_layer: Ash.DataLayer.Ets,
    extensions: [AshGraphql.Resource]

  require Ash.Query

  graphql do
    type :channel

    queries do
      get :channel, :read
    end
  end

  actions do
    create :create do
      primary?(true)
    end

    read(:read, primary?: true)

    update(:update, primary?: true)

    destroy(:destroy, primary?: true)
  end

  attributes do
    uuid_primary_key(:id)

    create_timestamp(:created_at, private?: false)
  end

  calculations do
    calculate(:direct_channel_messages, {:array, AshGraphql.Test.MessageUnion}, fn record,
                                                                                   %{
                                                                                     api: api
                                                                                   } ->
      record = api.load!(record, :messages)

      {:ok,
       record.messages
       |> Enum.map(&%Ash.Union{type: AshGraphql.Test.MessageUnion.struct_to_name(&1), value: &1})}
    end)

    calculate :indirect_channel_messages,
              AshGraphql.Test.PageOfChannelMessages,
              AshGraphql.Test.PageOfChannelMessagesCalculation do
      argument :offset, :integer do
        default(0)
      end

      argument :limit, :integer do
        default(10)
      end
    end
  end

  aggregates do
    count(:channel_message_count, :messages)
  end

  relationships do
    has_many(:messages, AshGraphql.Test.Message)
  end
end