Search code examples
elixirerlang-otp

Await an OTP process to exit


Suppose you have an OTP process whose completion you want to synchronously wait on (where "completion" may be a normal exit or a crash, stop, etc.).

Suppose further that for business reasons, you can't spawn this process with Task.async/1 or related Task utilities—it has to be a "normal" process not dependent on Task.await/2.

Is there a better way to do this than simply intermittently polling Process.alive?/1? This strikes me as the sort of thing there's probably an established pattern for, but for the life of me I can't find it.

def await(pid, timeout) when timeout <= 0 do
  if Process.alive?(pid) do
    Process.exit(pid, :kill) # (or however you want to handle timeouts)
  end
end

@await_sleep_ms 500
def await(pid, timeout) do
  # Is there a better way to do this?
  if Process.alive?(pid) do
    :timer.sleep(@await_sleep_ms)
    await(pid, subtract_timeout(timeout, @await_sleep_ms))
  end
end

Solution

  • The Process.monitor/1 function monitors the given process from the calling process. Combining this with receive, you can react to your process mailbox.

    defmodule Await do
      def spawn do
        Kernel.spawn(fn -> :timer.sleep(2000) end)
      end
      
      def await_down(pid) do
        ref = Process.monitor(pid)
        receive do
          {:DOWN, ^ref, :process, ^pid, _reason} = message -> {:ok, message}
        after
          3000 -> {:error, :timeout}
        end
      end
    end
    
    pid = Await.spawn()
    # => #PID<0.332.0>
    
    Await.await_down(pid)
    # => {:ok, {:DOWN, #Reference<0.4083290149.3661365255.180715>, :process, #PID<0.332.0>, :normal}}
    

    Note the pattern matching inside the receive block, to ensure not only the message is from your process, but that is comes from that specific monitoring.