Current section

Files

Jump to
commanded test event_store support subscriber.ex
Raw

test/event_store/support/subscriber.ex

defmodule Commanded.EventStore.Subscriber do
use GenServer
alias Commanded.EventStore
alias Commanded.EventStore.Subscriber
defmodule State do
defstruct [:owner, :subscription, received_events: [], subscribed?: false]
end
alias Subscriber.State
def start_link(owner) do
GenServer.start_link(__MODULE__, %State{owner: owner})
end
def init(%State{} = state) do
{:ok, subscription} = EventStore.subscribe_to(:all, "subscriber", self(), :origin)
{:ok, %State{state | subscription: subscription}}
end
def subscribed?(subscriber), do: GenServer.call(subscriber, :subscribed?)
def received_events(subscriber), do: GenServer.call(subscriber, :received_events)
def handle_call(:subscribed?, _from, %State{subscribed?: subscribed?} = state) do
{:reply, subscribed?, state}
end
def handle_call(:received_events, _from, %State{received_events: received_events} = state) do
{:reply, received_events, state}
end
def handle_info({:subscribed, subscription}, %State{subscription: subscription} = state) do
%State{owner: owner} = state
send(owner, {:subscribed, subscription})
{:noreply, %State{state | subscribed?: true}}
end
def handle_info({:events, events}, %State{} = state) do
%State{owner: owner, received_events: received_events, subscription: subscription} = state
send(owner, {:events, events})
state = %State{state | received_events: received_events ++ events}
EventStore.ack_event(subscription, List.last(events))
{:noreply, state}
end
end