X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fpleroma%2Fweb%2Fstreamer.ex;h=10670e71fbd343aac1f30e0ecedc8ba8ddad1e40;hb=1f32ba052c52ad974d17d8b65fb2c66b61eceffe;hp=a417178bae0e781f82d3af32dfdfafee620b6ae5;hpb=611ca385dea3d611a97579000311cc42684305e6;p=akkoma diff --git a/lib/pleroma/web/streamer.ex b/lib/pleroma/web/streamer.ex index a417178ba..10670e71f 100644 --- a/lib/pleroma/web/streamer.ex +++ b/lib/pleroma/web/streamer.ex @@ -5,9 +5,11 @@ defmodule Pleroma.Web.Streamer do def start_link do spawn(fn -> - Process.sleep(1000 * 30) # 30 seconds + # 30 seconds + Process.sleep(1000 * 30) GenServer.cast(__MODULE__, %{action: :ping}) end) + GenServer.start_link(__MODULE__, %{}, name: __MODULE__) end @@ -25,39 +27,54 @@ defmodule Pleroma.Web.Streamer do def handle_cast(%{action: :ping}, topics) do Map.values(topics) - |> List.flatten - |> Enum.each(fn (socket) -> + |> List.flatten() + |> Enum.each(fn socket -> Logger.debug("Sending keepalive ping") - send socket.transport_pid, {:text, ""} + send(socket.transport_pid, {:text, ""}) end) + spawn(fn -> - Process.sleep(1000 * 30) # 30 seconds + # 30 seconds + Process.sleep(1000 * 30) GenServer.cast(__MODULE__, %{action: :ping}) end) + {:noreply, topics} end def handle_cast(%{action: :stream, topic: "user", item: %Notification{} = item}, topics) do topic = "user:#{item.user_id}" - Enum.each(topics[topic] || [], fn (socket) -> - json = %{ - event: "notification", - payload: Pleroma.Web.MastodonAPI.MastodonAPIController.render_notification(socket.assigns["user"], item) |> Poison.encode! - } |> Poison.encode! - send socket.transport_pid, {:text, json} + Enum.each(topics[topic] || [], fn socket -> + json = + %{ + event: "notification", + payload: + Pleroma.Web.MastodonAPI.MastodonAPIController.render_notification( + socket.assigns["user"], + item + ) + |> Jason.encode!() + } + |> Jason.encode!() + + send(socket.transport_pid, {:text, json}) end) + {:noreply, topics} end def handle_cast(%{action: :stream, topic: "user", item: item}, topics) do Logger.debug("Trying to push to users") - recipient_topics = User.get_recipients_from_activity(item) - |> Enum.map(fn (%{id: id}) -> "user:#{id}" end) - Enum.each(recipient_topics, fn (topic) -> + recipient_topics = + User.get_recipients_from_activity(item) + |> Enum.map(fn %{id: id} -> "user:#{id}" end) + + Enum.each(recipient_topics, fn topic -> push_to_socket(topics, topic, item) end) + {:noreply, topics} end @@ -92,13 +109,21 @@ defmodule Pleroma.Web.Streamer do end def push_to_socket(topics, topic, item) do - Enum.each(topics[topic] || [], fn (socket) -> - json = %{ - event: "update", - payload: Pleroma.Web.MastodonAPI.StatusView.render("status.json", activity: item, for: socket.assigns[:user]) |> Poison.encode! - } |> Poison.encode! - - send socket.transport_pid, {:text, json} + Enum.each(topics[topic] || [], fn socket -> + json = + %{ + event: "update", + payload: + Pleroma.Web.MastodonAPI.StatusView.render( + "status.json", + activity: item, + for: socket.assigns[:user] + ) + |> Jason.encode!() + } + |> Jason.encode!() + + send(socket.transport_pid, {:text, json}) end) end