ielixir

defmodule IElixir.Socket.Heartbeat do
  @moduledoc """
  This is module responsible for handling Heartbeat requests.
  """

  use GenServer
  require Logger

  @doc false
  def start_link(opts) do
    GenServer.start_link(__MODULE__, opts, name: Heartbeat)
  end

  def init(opts) do
    Process.flag(:trap_exit, true)
    sock = IElixir.Utils.make_socket(opts, "hb", :rep)
    {:ok, sock}
  end

  def terminate(_reason, sock) do
    Logger.debug("Shutdown Heartbeat")
    :erlzmq.close(sock)
  end

  def handle_info({:zmq, _, data, []}, sock) do
    Logger.debug("Heartbeat ping received")
    :erlzmq.send(sock, data)
    {:noreply, sock}
  end
  def handle_info(msg, state) do
    Logger.warn("Got unexpected message on hb process: #{inspect msg}")
    {:noreply, state}
  end
end