fix(core): remove hang risk caused by coop budget in tokio (#8434)

Author: hardfist

crates/rspack_core/src/compiler/module_executor/mod.rs

@@ -12,6 +12,7 @@ use tokio::sync::{
   mpsc::{unbounded_channel, UnboundedSender},
   oneshot,
 };
+use tokio::task;
 
 use self::{
   ctrl::{CtrlTask, Event, ExecuteParam},
@@ -72,8 +73,9 @@ impl ModuleExecutor {
     let (stop_sender, stop_receiver) = oneshot::channel();
     self.event_sender = Some(event_sender.clone());
     self.stop_receiver = Some(stop_receiver);
-
-    tokio::spawn(async move {
+    // avoid coop budget consumed to zero cause hang risk
+    // related to https://tokio.rs/blog/2020-04-preemption
+    tokio::spawn(task::unconstrained(async move {
       let _ = run_task_loop_with_event(
         &mut ctx,
         vec![Box::new(CtrlTask::new(event_receiver))],
@@ -89,7 +91,7 @@ impl ModuleExecutor {
       stop_sender
         .send(ctx.transform_to_make_artifact())
         .expect("should success");
-    });
+    }));
   }
 
   pub async fn hook_after_finish_modules(&mut self, compilation: &mut Compilation) {

crates/rspack_core/src/utils/task_loop.rs

@@ -10,7 +10,10 @@ use std::{
 
 use rspack_error::Result;
 use rspack_util::ext::AsAny;
-use tokio::sync::mpsc::{self, error::TryRecvError};
+use tokio::{
+  sync::mpsc::{self, error::TryRecvError},
+  task,
+};
 
 /// Result returned by task
 ///
@@ -81,12 +84,12 @@ pub async fn run_task_loop_with_event<Ctx: 'static>(
           let tx = tx.clone();
           let is_expected_shutdown = is_expected_shutdown.clone();
           active_task_count += 1;
-          tokio::spawn(async move {
+          tokio::spawn(task::unconstrained(async move {
             let r = task.background_run().await;
             if !is_expected_shutdown.load(Ordering::Relaxed) {
               tx.send(r).expect("failed to send error message");
             }
-          });
+          }));
         }
         TaskType::Sync => {
           // merge sync task result directly