doc tweak

pma · Mar 8, 2021 · 20b20a9 · 20b20a9
1 parent 42bf0b8
commit 20b20a9
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lib/amqp/application.ex b/lib/amqp/application.ex
@@ -210,13 +210,17 @@ defmodule AMQP.Application do
         ....
 
         def handle_info(:subscribe, state) do
-          subscribe()
-          {:noreply, state}
+          case subscribe() do
+            {:ok, chan} ->
+              {:noreply, Map.put(state, :channel, chan)}
+            _ ->
+              {:noreply, state}
+          end
         end
 
-        def handle_info({:DOWN, _, :process, pid, reason}, state) do
+        def handle_info({:DOWN, _, :process, pid, reason}, %{channel: %{pid: pid}} = state) do
           send(self(), :subscribe)
-          {:noreply, state}
+          {:noreply, Map.put(state, :channel, nil)}
         end
 
         defp subscribe() do