livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-02-03 04:38:11 +08:00
Code Issues Projects Releases Packages Wiki Activity

Ignore channel events for unknown JS Views (#2363)

Browse source
This commit is contained in:
Jonatan Kłosko 2023-11-17 13:10:48 +01:00 committed by GitHub
parent d345c9eeff
commit 9076e15a69
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 32 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
lib/livebook_web/channels/js_view_channel.ex
View file

@ -47,30 +47,40 @@ defmodule LivebookWeb.JSViewChannel do
def handle_in("event", raw, socket) do def handle_in("event", raw, socket) do
{[event, ref], payload} = transport_decode!(raw) {[event, ref], payload} = transport_decode!(raw)
pid = socket.assigns.ref_with_info[ref].pid
send(pid, {:event, event, payload, %{origin: socket.assigns.client_id, ref: ref}}) with %{^ref => info} <- socket.assigns.ref_with_info do
send(info.pid, {:event, event, payload, %{origin: socket.assigns.client_id, ref: ref}})
end
{:noreply, socket} {:noreply, socket}
end end
def handle_in("ping", %{"ref" => ref}, socket) do def handle_in("ping", %{"ref" => ref}, socket) do
pid = socket.assigns.ref_with_info[ref].pid with %{^ref => info} <- socket.assigns.ref_with_info do
send(pid, {:ping, self(), nil, %{ref: ref}}) send(info.pid, {:ping, self(), nil, %{ref: ref}})
end
{:noreply, socket} {:noreply, socket}
end end
def handle_in("disconnect", %{"ref" => ref}, socket) do def handle_in("disconnect", %{"ref" => ref}, socket) do
socket = socket =
if socket.assigns.ref_with_info[ref].count == 1 do case socket.assigns.ref_with_info do
Livebook.Session.unsubscribe_from_runtime_events( %{^ref => %{count: 1}} ->
socket.assigns.session_id, Livebook.Session.unsubscribe_from_runtime_events(
"js_live", socket.assigns.session_id,
ref "js_live",
) ref
)
{_, socket} = pop_in(socket.assigns.ref_with_info[ref]) {_, socket} = pop_in(socket.assigns.ref_with_info[ref])
socket socket
else
update_in(socket.assigns.ref_with_info[ref], &%{&1 | count: &1.count - 1}) %{^ref => %{count: count}} when count > 1 ->
update_in(socket.assigns.ref_with_info[ref], &%{&1 | count: &1.count - 1})
%{} ->
socket
end end
{:noreply, socket} {:noreply, socket}

8
test/livebook_web/channels/js_view_channel_test.exs
View file

@ -57,6 +57,14 @@ defmodule LivebookWeb.JSViewChannelTest do
assert_push "event:1", %{"root" => [["ping"], [1, 2, 3]]} assert_push "event:1", %{"root" => [["ping"], [1, 2, 3]]}
end end
test "ignores client events when no connection is found", %{socket: socket} do
push(socket, "event", %{"root" => [["ping", "1"], [1, 2, 3]]})
# The channel should still be operational
push(socket, "connect", %{"connect_token" => connect_token(), "ref" => "1", "id" => "id1"})
assert_receive {:connect, _from, %{}}
end
describe "binary payload" do describe "binary payload" do
test "initial data", %{socket: socket} do test "initial data", %{socket: socket} do
push(socket, "connect", %{"connect_token" => connect_token(), "ref" => "1", "id" => "id1"}) push(socket, "connect", %{"connect_token" => connect_token(), "ref" => "1", "id" => "id1"})