Label DbConnection's own processes

Author: nathanl

lib/db_connection/connection.ex

@@ -48,6 +48,10 @@ defmodule DBConnection.Connection do
   @doc false
   @impl :gen_statem
   def init({mod, opts, pool, tag}) do
+    pool_index = Keyword.get(opts, :pool_index)
+    label = if pool_index, do: "db_conn_#{pool_index}", else: "db_conn"
+    Util.set_label(label)
+
     s = %{
       mod: mod,
       opts: opts,

lib/db_connection/ownership/proxy.ex

@@ -22,6 +22,8 @@ defmodule DBConnection.Ownership.Proxy do
 
   @impl true
   def init({caller, pool, pool_opts}) do
+    Util.set_label("db_ownership_proxy")
+
     pool_opts =
       pool_opts
       |> Keyword.put(:timeout, :infinity)

lib/db_connection/task.ex

@@ -13,6 +13,8 @@ defmodule DBConnection.Task do
   end
 
   def init(fun, parent, opts) do
+    DBConnection.Util.set_label("db_after_connect_task")
+
     try do
       Process.link(parent)
     catch

lib/db_connection/util.ex

@@ -13,6 +13,18 @@ defmodule DBConnection.Util do
     end
   end
 
+  @doc """
+  Set a process label if `Process.set_label/1` is available.
+  """
+  def set_label(label) do
+    if function_exported?(Process, :set_label, 1) do
+      Process.set_label(label)
+    else
+      :ok
+    end
+  end
+
+  # Get a process label if `:proc_lib.get_label/1` is available.
   defp get_label(pid) do
     if function_exported?(:proc_lib, :get_label, 1) do
       # Avoid a compiler warning if the function isn't