propagate all collectibles

vercel · Nov 12, 2024 · 79bba37 · 79bba37
1 parent ed72ab8
commit 79bba37
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 24 deletions.
diff --git a/turbopack/crates/turbo-tasks-backend/src/backend/operation/aggregation_update.rs b/turbopack/crates/turbo-tasks-backend/src/backend/operation/aggregation_update.rs
@@ -146,7 +146,7 @@ impl AggregatedDataUpdate {
         let aggregation = get_aggregation_number(task);
         let mut dirty_container_count = Default::default();
         let mut collectibles_update: Vec<_> =
-            get_many!(task, Collectible { collectible } count if *count > 0 => (*collectible, 1));
+            get_many!(task, Collectible { collectible } count => (*collectible, *count));
         if is_aggregating_node(aggregation) {
             dirty_container_count = get!(task, AggregatedDirtyContainerCount)
                 .cloned()

diff --git a/turbopack/crates/turbo-tasks-backend/src/backend/operation/cleanup_old_edges.rs b/turbopack/crates/turbo-tasks-backend/src/backend/operation/cleanup_old_edges.rs
@@ -109,23 +109,14 @@ impl Operation for CleanupOldEdgesOperation {
                                 });
                                 let mut task = ctx.task(task_id, TaskDataCategory::All);
                                 for (collectible, count) in collectibles.iter_mut() {
-                                    if update_count!(
+                                    update_count!(
                                         task,
                                         Collectible {
                                             collectible: *collectible
                                         },
                                         *count
-                                    ) {
-                                        if *count > 0 {
-                                            *count = 1;
-                                        } else {
-                                            *count = -1;
-                                        }
-                                    } else {
-                                        *count = 0;
-                                    }
+                                    );
                                 }
-                                collectibles.retain(|(_, count)| *count != 0);
                                 queue.extend(AggregationUpdateJob::data_update(
                                     &mut task,
                                     AggregatedDataUpdate::new().collectibles_update(collectibles),

diff --git a/turbopack/crates/turbo-tasks-backend/src/backend/operation/update_collectible.rs b/turbopack/crates/turbo-tasks-backend/src/backend/operation/update_collectible.rs
@@ -43,18 +43,12 @@ impl UpdateCollectibleOperation {
                 // Not reduced from outdated
             }
         }
-        if count != 0 && update_count!(task, Collectible { collectible }, count) {
-            if count > 0 {
-                queue.extend(AggregationUpdateJob::data_update(
-                    &mut task,
-                    AggregatedDataUpdate::new().collectibles_update(vec![(collectible, 1)]),
-                ));
-            } else {
-                queue.extend(AggregationUpdateJob::data_update(
-                    &mut task,
-                    AggregatedDataUpdate::new().collectibles_update(vec![(collectible, -1)]),
-                ));
-            }
+        if count != 0 {
+            update_count!(task, Collectible { collectible }, count);
+            queue.extend(AggregationUpdateJob::data_update(
+                &mut task,
+                AggregatedDataUpdate::new().collectibles_update(vec![(collectible, count)]),
+            ));
         }
 
         drop(task);