Current section

Files

Jump to
eventstore lib event_store registration local_registry.ex
Raw

lib/event_store/registration/local_registry.ex

defmodule EventStore.Registration.LocalRegistry do
@moduledoc """
Pub/sub using Elixir's local `Registry` module, restricted to running on a
single node only.
"""
@behaviour EventStore.Registration
@doc """
Return the local supervisor child spec.
"""
@spec child_spec() :: [:supervisor.child_spec()]
@impl EventStore.Registration
def child_spec do
[
Supervisor.child_spec(
{
Registry,
keys: :duplicate, name: EventStore.PubSub, partitions: System.schedulers_online()
},
id: EventStore.PubSub
)
]
end
@doc """
Subscribes the caller to the given topic.
"""
@spec subscribe(binary, mapper: (RecordedEvent.t() -> any())) :: :ok | {:error, term}
@impl EventStore.Registration
def subscribe(topic, opts) do
with {:ok, _} <- Registry.register(EventStore.PubSub, topic, opts) do
:ok
end
end
@doc """
Broadcasts message on given topic.
"""
@spec broadcast(binary, term) :: :ok | {:error, term}
@impl EventStore.Registration
def broadcast(topic, message) do
Registry.dispatch(EventStore.PubSub, topic, fn entries ->
for {pid, opts} <- entries do
notify_subscriber(pid, message, opts)
end
end)
end
defp notify_subscriber(pid, {:events, events}, mapper: mapper) when is_function(mapper, 1) do
send(pid, {:events, Enum.map(events, mapper)})
end
defp notify_subscriber(pid, message, _opts) do
send(pid, message)
end
end