tweak child_specs

lib/nsq/connection.ex

@@ -6,6 +6,7 @@ defmodule NSQ.Connection do
   # ------------------------------------------------------- #
   # Directives                                              #
   # ------------------------------------------------------- #
+  use GenServer
   alias NSQ.Connection.Command
   alias NSQ.Connection.Initializer
   alias NSQ.Connection.MessageHandling
@@ -62,18 +63,15 @@ defmodule NSQ.Connection do
   # ------------------------------------------------------- #
   # Behaviour Implementation                                #
   # ------------------------------------------------------- #
-  @spec start_link(pid, host_with_port, NSQ.Config.t(), String.t(), String.t(), pid, list) ::
+  @spec start_link(pid | host_with_port | NSQ.Config.t() | String.t() | String.t() | pid) ::
           {:ok, pid}
-  def start_link(
-        parent,
-        nsqd,
-        config,
-        topic,
-        channel,
-        conn_info_pid,
-        event_manager_pid,
-        opts \\ []
-      ) do
+  def start_link([parent, nsqd, config, topic, channel, conn_info_pid, event_manager_pid]),
+    do: start_link([parent, nsqd, config, topic, channel, conn_info_pid, event_manager_pid, []])
+
+  @spec start_link(pid | host_with_port | NSQ.Config.t() | String.t() | String.t() | pid | list) ::
+          {:ok, pid}
+
+  def start_link([parent, nsqd, config, topic, channel, conn_info_pid, event_manager_pid, opts]) do
     state = %{
       @initial_state
       | parent: parent,

lib/nsq/connection/supervisor.ex

@@ -36,10 +36,9 @@ defmodule NSQ.Connection.Supervisor do
 
     # When using nsqlookupd, we expect connections will be naturally
     # rediscovered if they fail.
-    config =
-      [id: conn_id, start: {NSQ.Connection, :start_link, args}, restart: :temporary] ++ opts
+    opts = [id: conn_id, restart: :temporary] ++ opts
 
-    child = Map.new(config)
+    child = Supervisor.child_spec({NSQ.Connection, args}, opts)
 
     Supervisor.start_child(conn_sup_pid, child)
   end

lib/nsq/consumer.ex

@@ -131,8 +131,11 @@ defmodule NSQ.Consumer do
   @doc """
   Starts a Consumer process, called via the supervisor.
   """
-  @spec start_link(String.t(), String.t(), NSQ.Config.t(), list) :: {:ok, pid}
-  def start_link(topic, channel, config, opts \\ []) do
+  @spec start_link(String.t() | String.t() | NSQ.Config.t()) :: {:ok, pid}
+  def start_link([topic, channel, config]), do: start_link([topic, channel, config, []])
+
+  @spec start_link(String.t() | String.t() | NSQ.Config.t() | list) :: {:ok, pid}
+  def start_link([topic, channel, config, opts]) do
     {:ok, config} = NSQ.Config.validate(config)
     {:ok, config} = NSQ.Config.normalize(config)
     unless is_valid_topic_name?(topic), do: raise("Invalid topic name #{topic}")

lib/nsq/consumer/supervisor.ex

@@ -12,10 +12,7 @@ defmodule NSQ.Consumer.Supervisor do
     rdy_loop_id = String.to_atom("#{consumer_name}_rdy_loop")
 
     children = [
-      %{
-        id: NSQ.Consumer,
-        start: {NSQ.Consumer, :start_link, [topic, channel, config, [name: consumer_name]]}
-      },
+      {NSQ.Consumer, [topic, channel, config, [name: consumer_name]]},
       Supervisor.child_spec(
         {Task, fn -> NSQ.Consumer.Connections.discovery_loop(consumer_name) end},
         id: discovery_loop_id

lib/nsq/message/supervisor.ex

@@ -17,8 +17,8 @@ defmodule NSQ.Message.Supervisor do
   def start_child(msg_sup_pid, message, opts \\ []) do
     # If a message fails, NSQ will handle requeueing.
     id = message.id <> "-" <> UUID.uuid4(:hex)
-    config = [id: id, start: {NSQ.Message, :start_link, [message]}, restart: :temporary] ++ opts
-    child = Map.new(config)
+    opts = [id: id, restart: :temporary] ++ opts
+    child = Supervisor.child_spec({NSQ.Message, message}, opts)
     Supervisor.start_child(msg_sup_pid, child)
   end
 

lib/nsq/producer.ex

@@ -134,8 +134,11 @@ defmodule NSQ.Producer do
   # ------------------------------------------------------- #
   # API Definitions                                         #
   # ------------------------------------------------------- #
-  @spec start_link(binary, NSQ.Config.t(), GenServer.options()) :: {:ok, pid}
-  def start_link(topic, config, genserver_options \\ []) do
+  @spec start_link([{binary | NSQ.Config.t()}]) :: {:ok, pid}
+  def start_link([topic, config]), do: start_link([topic, config, []])
+
+  @spec start_link([binary | NSQ.Config.t() | GenServer.options()]) :: {:ok, pid}
+  def start_link([topic, config, genserver_options]) do
     {:ok, config} = NSQ.Config.validate(config || %NSQ.Config{})
     {:ok, config} = NSQ.Config.normalize(config)
     unless is_valid_topic_name?(topic), do: raise("Invalid topic name #{topic}")

lib/nsq/producer/supervisor.ex

@@ -7,7 +7,7 @@ defmodule NSQ.Producer.Supervisor do
 
   @impl true
   def init({topic, config}) do
-    children = [%{id: NSQ.Producer, start: {NSQ.Producer, :start_link, [topic, config]}}]
+    children = [{NSQ.Producer, [topic, config]}]
 
     Supervisor.init(children, strategy: :one_for_one)
   end