Notify receiver on client disconnection (#9)

Author: gBillal

lib/ex_rtmp/client.ex

@@ -157,7 +157,7 @@ defmodule ExRTMP.Client do
 
   @impl true
   def handle_info({:tcp_closed, _port}, state) do
-    Logger.info("RTMP connection closed")
+    send(state.receiver, {:disconnected, self()})
     {:noreply, State.reset(state)}
   end
 

lib/ex_rtmp/server.ex

@@ -109,7 +109,7 @@ defmodule ExRTMP.Server do
 
   @impl true
   def handle_call(:stop, _from, state) do
-    Process.exit(state.listener, :normal)
+    Process.exit(state.listener, :kill)
     {:stop, :normal, :ok, state}
   end
 
@@ -136,7 +136,7 @@ defmodule ExRTMP.Server do
         Logger.debug("New client connected")
 
         {:ok, pid} =
-          ClientSession.start(
+          ClientSession.start_link(
             socket: client_socket,
             handler: state.handler,
             handler_options: state.handler_options,

lib/ex_rtmp/server/client_session.ex

@@ -50,6 +50,11 @@ defmodule ExRTMP.Server.ClientSession do
     GenServer.start(__MODULE__, opts)
   end
 
+  @spec start_link(keyword()) :: GenServer.on_start()
+  def start_link(opts) do
+    GenServer.start_link(__MODULE__, opts)
+  end
+
   @doc """
   Sends video data to the client.
   """

test/ex_rtmp/client_test.exs

@@ -59,6 +59,9 @@ defmodule ExRTMP.ClientTest do
 
       assert expected_access_units == collected_access_units
 
+      ExRTMP.Server.stop(server)
+      assert_receive {:disconnected, ^pid}, 2000
+
       Client.stop(pid)
     end
   end