task: quickly send task to heap on debug mode (#4009)

tokio-rs · Aug 12, 2021 · 84c4a6d · 84c4a6d · github-actions · Aug 12, 2021
1 parent b501f25
commit 84c4a6d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tokio/src/task/spawn.rs b/tokio/src/task/spawn.rs
@@ -127,7 +127,13 @@ cfg_rt! {
         T: Future + Send + 'static,
         T::Output: Send + 'static,
     {
-        spawn_inner(future, None)
+        // preventing stack overflows on debug mode, by quickly sending the
+        // task to the heap.
+        if cfg!(debug_assertions) && std::mem::size_of::<T>() > 2048 {
+            spawn_inner(Box::pin(future), None)
+        } else {
+            spawn_inner(future, None)
+        }
     }
 
     #[cfg_attr(tokio_track_caller, track_caller)]