1 defmodule Pleroma.Web.Streamer.State do
5 alias Pleroma.Web.Streamer.StreamerSocket
8 GenServer.start_link(__MODULE__, %{sockets: %{}}, name: __MODULE__)
11 def add_socket(topic, socket) do
12 GenServer.call(__MODULE__, {:add, socket, topic})
15 def remove_socket(topic, socket) do
16 GenServer.call(__MODULE__, {:remove, socket, topic})
20 %{sockets: stream_sockets} = GenServer.call(__MODULE__, :get_state)
28 def handle_call(:get_state, _from, state) do
29 {:reply, state, state}
32 def handle_call({:add, socket, topic}, _from, %{sockets: sockets} = state) do
33 internal_topic = internal_topic(topic, socket)
34 stream_socket = StreamerSocket.from_socket(socket)
38 |> Map.get(internal_topic, [])
39 |> List.insert_at(0, stream_socket)
42 state = put_in(state, [:sockets, internal_topic], sockets_for_topic)
43 Logger.debug("Got new conn for #{topic}")
44 {:reply, state, state}
47 def handle_call({:remove, socket, topic}, _from, %{sockets: sockets} = state) do
48 internal_topic = internal_topic(topic, socket)
49 stream_socket = StreamerSocket.from_socket(socket)
53 |> Map.get(internal_topic, [])
54 |> List.delete(stream_socket)
56 state = Kernel.put_in(state, [:sockets, internal_topic], sockets_for_topic)
57 {:reply, state, state}
60 defp internal_topic(topic, socket)
61 when topic in ~w[user user:notification direct] do
62 "#{topic}:#{socket.assigns[:user].id}"
65 defp internal_topic(topic, _) do