Improve crash handling

v0idpwn · Apr 6, 2023 · a0b3d1f · a0b3d1f
1 parent 1546594
commit a0b3d1f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/kafkaesque/pipeline.ex b/lib/kafkaesque/pipeline.ex
@@ -18,6 +18,7 @@ defmodule Kafkaesque.Pipeline do
     GenServer.start_link(__MODULE__, opts)
   end
 
+  @impl GenServer
   def init(opts) do
     Process.flag(:trap_exit, true)
 
@@ -41,6 +42,12 @@ defmodule Kafkaesque.Pipeline do
      }}
   end
 
+  @impl GenServer
+  def handle_info({:EXIT, _pid, reason}, state) do
+    {:stop, reason, state}
+  end
+
+  @impl GenServer
   def terminate(:shutdown, state) do
     # Drain
     :ok = Producer.stop_producing(state.producer)