fix task id reuse

vercel · Aug 8, 2024 · 423146d · 423146d
1 parent 8f7f100
commit 423146d
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/turbopack/crates/turbo-tasks-memory/src/memory_backend.rs b/turbopack/crates/turbo-tasks-memory/src/memory_backend.rs
@@ -26,8 +26,8 @@ use turbo_tasks::{
     },
     event::EventListener,
     util::{IdFactoryWithReuse, NoMoveVec},
-    CellId, RawVc, TaskId, TaskIdSet, TraitTypeId, TurboTasksBackendApi, Unused, ValueTypeId,
-    TRANSIENT_TASK_BIT,
+    CellId, RawVc, TaskId, TaskIdSet, TaskInput, TraitTypeId, TurboTasksBackendApi, Unused,
+    ValueTypeId, TRANSIENT_TASK_BIT,
 };
 
 use crate::{
@@ -272,8 +272,13 @@ impl MemoryBackend {
                 drop(entry);
                 unsafe {
                     task_storage.remove(index);
-                    let new_id = Unused::new_unchecked(new_id);
-                    turbo_tasks.reuse_persistent_task_id(new_id);
+                    if new_id.is_transient() {
+                        let new_id = Unused::new_unchecked(new_id);
+                        turbo_tasks.reuse_transient_task_id(new_id);
+                    } else {
+                        let new_id = Unused::new_unchecked(new_id);
+                        turbo_tasks.reuse_persistent_task_id(new_id);
+                    }
                 }
                 task_id
             }