Ensure module is loaded (#30)

dashbitco · Feb 26, 2024 · 4f6aaae · 4f6aaae
1 parent 17a36c9
commit 4f6aaae
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lib/nimble_pool.ex b/lib/nimble_pool.ex
@@ -501,7 +501,15 @@ defmodule NimblePool do
   @impl true
   def init({worker, arg, pool_size, lazy, worker_idle_timeout, max_idle_pings}) do
     Process.flag(:trap_exit, true)
-    _ = Code.ensure_loaded(worker)
+
+    case Code.ensure_loaded(worker) do
+      {:module, _} ->
+        :ok
+
+      {:error, reason} ->
+        raise ArgumentError, "failed to load worker module #{inspect(worker)}: #{inspect(reason)}"
+    end
+
     lazy = if lazy, do: pool_size, else: nil
 
     if worker_idle_timeout do