Disconnect TCP connection on EOS (#4)

membraneframework · Mar 18, 2024 · 09f4114 · 09f4114
1 parent 474d9fd
commit 09f4114
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -11,7 +11,7 @@ This package provides TCP Source and Sink, that read and write to TCP sockets.
 Add the following line to your `deps` in `mix.exs`. Run `mix deps.get`.
 
 ```elixir
-	{:membrane_tcp_plugin, "~> 0.2.0"}
+	{:membrane_tcp_plugin, "~> 0.3.0"}
 ```
 
 ## Copyright and License

diff --git a/lib/membrane_tcp/sink.ex b/lib/membrane_tcp/sink.ex
@@ -82,4 +82,10 @@ defmodule Membrane.TCP.Sink do
 
   @impl true
   defdelegate handle_setup(context, state), to: CommonSocketBehaviour
+
+  @impl true
+  def handle_end_of_stream(_pad, _context, state) do
+    local_socket = Socket.close(state.local_socket)
+    {[], %{state | local_socket: local_socket}}
+  end
 end
diff --git a/mix.exs b/mix.exs
@@ -1,7 +1,7 @@
 defmodule Membrane.TCP.MixProject do
   use Mix.Project
 
-  @version "0.2.0"
+  @version "0.3.0"
   @github_url "https://github.com/membraneframework/membrane_tcp_plugin"
 
   def project do