From 8d1cc5149155959e01d7c85851d43a6691a47a49 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Valim?= <jose.valim@dashbit.co>
Date: Wed, 20 Nov 2024 17:22:55 +0100
Subject: [PATCH] Load modules for mix recompile in parallel

---
 lib/mix/lib/mix/compilers/elixir.ex | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/lib/mix/lib/mix/compilers/elixir.ex b/lib/mix/lib/mix/compilers/elixir.ex
index 019e155de9d..2d882f6eb72 100644
--- a/lib/mix/lib/mix/compilers/elixir.ex
+++ b/lib/mix/lib/mix/compilers/elixir.ex
@@ -373,8 +373,7 @@ defmodule Mix.Compilers.Elixir do
             Map.has_key?(stale_modules, module) ->
               {[module | modules_to_recompile], modules_to_mix_check}
 
-            recompile? and Code.ensure_loaded?(module) and
-                function_exported?(module, :__mix_recompile__?, 0) ->
+            recompile? ->
               {modules_to_recompile, [module | modules_to_mix_check]}
 
             true ->
@@ -385,7 +384,17 @@ defmodule Mix.Compilers.Elixir do
     modules_to_recompile =
       modules_to_recompile ++
         for {:ok, {module, true}} <-
-              Task.async_stream(modules_to_mix_check, &{&1, &1.__mix_recompile__?()},
+              Task.async_stream(
+                modules_to_mix_check,
+                fn mod ->
+                  # Since loading the modules themselves can be expensive,
+                  # we want to do it inside the task too. Modules being deleted
+                  # are uncommon, so this optimizes the common case.
+                  {mod,
+                   Code.ensure_loaded?(mod) and
+                     function_exported?(mod, :__mix_recompile__?, 0) and
+                     mod.__mix_recompile__?()}
+                end,
                 ordered: false,
                 timeout: :infinity
               ) do