Skip to content

Commit a654851

Browse files
sokrasokra
committed
fix hanging waiting on in progress cells
1 parent 55f9fd1 commit a654851

File tree

6 files changed

+266
-46
lines changed

6 files changed

+266
-46
lines changed

‎turbopack/crates/turbo-tasks-auto-hash-map/src/lib.rs‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
#![feature(hash_raw_entry)]
2+
#![feature(hash_extract_if)]
23

34
pub mod map;
45
pub mod set;

‎turbopack/crates/turbo-tasks-auto-hash-map/src/map.rs‎

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,16 @@ impl<K: Eq + Hash, V, H: BuildHasher + Default, const I: usize> AutoMap<K, V, H,
235235
}
236236
}
237237

238+
pub fn extract_if<'l, F>(&'l mut self, f: F) -> ExtractIfIter<'l, K, V, I, F>
239+
where
240+
F: for<'a, 'b> FnMut(&'a K, &'b mut V) -> bool,
241+
{
242+
match self {
243+
AutoMap::List(list) => ExtractIfIter::List { list, index: 0, f },
244+
AutoMap::Map(map) => ExtractIfIter::Map(map.extract_if(f)),
245+
}
246+
}
247+
238248
/// see [HashMap::shrink_to_fit](https://doc.rust-lang.org/std/collections/struct.HashMap.html#method.shrink_to_fit)
239249
pub fn shrink_to_fit(&mut self) {
240250
match self {
@@ -843,6 +853,43 @@ where
843853
}
844854
}
845855

856+
pub enum ExtractIfIter<'l, K, V, const I: usize, F>
857+
where
858+
F: for<'a, 'b> FnMut(&'a K, &'b mut V) -> bool,
859+
{
860+
List {
861+
list: &'l mut SmallVec<[(K, V); I]>,
862+
index: usize,
863+
f: F,
864+
},
865+
Map(std::collections::hash_map::ExtractIf<'l, K, V, F>),
866+
}
867+
868+
impl<'l, K, V, const I: usize, F> Iterator for ExtractIfIter<'l, K, V, I, F>
869+
where
870+
F: for<'a, 'b> FnMut(&'a K, &'b mut V) -> bool,
871+
{
872+
type Item = (K, V);
873+
874+
fn next(&mut self) -> Option<Self::Item> {
875+
match self {
876+
ExtractIfIter::List { list, index, f } => {
877+
while *index < list.len() {
878+
let (key, value) = &mut list[*index];
879+
if f(key, value) {
880+
let item = list.swap_remove(*index);
881+
return Some(item);
882+
} else {
883+
*index += 1;
884+
}
885+
}
886+
None
887+
}
888+
ExtractIfIter::Map(extract_if) => extract_if.next(),
889+
}
890+
}
891+
}
892+
846893
#[cfg(test)]
847894
mod tests {
848895
use super::*;
@@ -863,4 +910,37 @@ mod tests {
863910
}
864911
assert_eq!(map.remove(&(MAX_LIST_SIZE * 2)), None);
865912
}
913+
914+
#[test]
915+
fn test_extract_if_map() {
916+
let mut map = AutoMap::new();
917+
for i in 0..MAX_LIST_SIZE * 2 {
918+
map.insert(i, i);
919+
}
920+
let iter = map.extract_if(|_, v| *v % 2 == 0);
921+
assert_eq!(iter.count(), MAX_LIST_SIZE);
922+
assert_eq!(map.len(), MAX_LIST_SIZE);
923+
}
924+
925+
#[test]
926+
fn test_extract_if_list() {
927+
let mut map = AutoMap::new();
928+
for i in 0..MIN_HASH_SIZE {
929+
map.insert(i, i);
930+
}
931+
let iter = map.extract_if(|_, v| *v % 2 == 0);
932+
assert_eq!(iter.count(), MIN_HASH_SIZE / 2);
933+
assert_eq!(map.len(), MIN_HASH_SIZE / 2);
934+
}
935+
936+
#[test]
937+
fn test_extract_if_list2() {
938+
let mut map = AutoMap::new();
939+
for i in 0..MIN_HASH_SIZE {
940+
map.insert(i, i);
941+
}
942+
let iter = map.extract_if(|_, v| *v < 5);
943+
assert_eq!(iter.count(), 5);
944+
assert_eq!(map.len(), MIN_HASH_SIZE - 5);
945+
}
866946
}

‎turbopack/crates/turbo-tasks-backend/src/backend/mod.rs‎

Lines changed: 77 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -604,7 +604,7 @@ impl<B: BackingStorage> TurboTasksBackendInner<B> {
604604
ctx.get_task_description(task_id)
605605
);
606606
};
607-
if cell.index > *max_id {
607+
if cell.index >= *max_id {
608608
add_cell_dependency(self, task, reader, cell, task_id, &mut ctx);
609609
bail!(
610610
"Cell {cell:?} no longer exists in task {} (index out of bounds)",
@@ -1228,7 +1228,6 @@ impl<B: BackingStorage> TurboTasksBackendInner<B> {
12281228
let _ = stateful;
12291229

12301230
// handle cell counters: update max index and remove cells that are no longer used
1231-
let mut removed_cells = HashMap::new();
12321231
let mut old_counters: HashMap<_, _> =
12331232
get_many!(task, CellTypeMaxIndex { cell_type } max_index => (*cell_type, *max_index));
12341233
for (&cell_type, &max_index) in cell_counters.iter() {
@@ -1238,9 +1237,6 @@ impl<B: BackingStorage> TurboTasksBackendInner<B> {
12381237
cell_type,
12391238
value: max_index,
12401239
});
1241-
if old_max_index > max_index {
1242-
removed_cells.insert(cell_type, max_index + 1..=old_max_index);
1243-
}
12441240
}
12451241
} else {
12461242
task.add_new(CachedDataItem::CellTypeMaxIndex {
@@ -1249,28 +1245,46 @@ impl<B: BackingStorage> TurboTasksBackendInner<B> {
12491245
});
12501246
}
12511247
}
1252-
for (cell_type, old_max_index) in old_counters {
1248+
for (cell_type, _) in old_counters {
12531249
task.remove(&CachedDataItemKey::CellTypeMaxIndex { cell_type });
1254-
removed_cells.insert(cell_type, 0..=old_max_index);
12551250
}
1251+
12561252
let mut removed_data = Vec::new();
1257-
for (&cell_type, range) in removed_cells.iter() {
1258-
for index in range.clone() {
1259-
removed_data.extend(
1260-
task.remove(&CachedDataItemKey::CellData {
1261-
cell: CellId {
1262-
type_id: cell_type,
1263-
index,
1264-
},
1265-
})
1266-
.into_iter(),
1267-
);
1268-
}
1269-
}
1253+
let mut old_edges = Vec::new();
12701254

1255+
// Remove no longer existing cells and notify in progress cells
12711256
// find all outdated data items (removed cells, outdated edges)
1272-
let old_edges = if task.is_indexed() {
1273-
let mut old_edges = Vec::new();
1257+
if task.is_indexed() {
1258+
removed_data.extend(task.extract_if(
1259+
CachedDataItemIndex::InProgressCell,
1260+
|key, value| {
1261+
match (key, value) {
1262+
(
1263+
&CachedDataItemKey::InProgressCell { cell },
1264+
CachedDataItemValue::InProgressCell { value },
1265+
) if cell_counters
1266+
.get(&cell.type_id)
1267+
.map_or(true, |start_index| cell.index >= *start_index) =>
1268+
{
1269+
value.event.notify(usize::MAX);
1270+
true
1271+
}
1272+
_ => false,
1273+
}
1274+
},
1275+
));
1276+
removed_data.extend(task.extract_if(CachedDataItemIndex::CellData, |key, _| {
1277+
match key {
1278+
&CachedDataItemKey::CellData { cell }
1279+
if cell_counters
1280+
.get(&cell.type_id)
1281+
.map_or(true, |start_index| cell.index >= *start_index) =>
1282+
{
1283+
true
1284+
}
1285+
_ => false,
1286+
}
1287+
}));
12741288
if self.should_track_children() {
12751289
old_edges.extend(task.iter(CachedDataItemIndex::Children).filter_map(
12761290
|(key, _)| match *key {
@@ -1306,9 +1320,9 @@ impl<B: BackingStorage> TurboTasksBackendInner<B> {
13061320
|(key, _)| {
13071321
match *key {
13081322
CachedDataItemKey::CellDependent { cell, task }
1309-
if removed_cells
1323+
if cell_counters
13101324
.get(&cell.type_id)
1311-
.map_or(false, |range| range.contains(&cell.index)) =>
1325+
.map_or(true, |start_index| cell.index >= *start_index) =>
13121326
{
13131327
Some(OutdatedEdge::RemovedCellDependent(task, cell.type_id))
13141328
}
@@ -1317,36 +1331,53 @@ impl<B: BackingStorage> TurboTasksBackendInner<B> {
13171331
},
13181332
));
13191333
}
1320-
old_edges
13211334
} else {
1322-
task.iter_all()
1323-
.filter_map(|(key, value)| match *key {
1324-
CachedDataItemKey::OutdatedChild { task } => Some(OutdatedEdge::Child(task)),
1325-
CachedDataItemKey::OutdatedCollectible { collectible } => {
1326-
let CachedDataItemValue::OutdatedCollectible { value } = *value else {
1327-
unreachable!();
1328-
};
1329-
Some(OutdatedEdge::Collectible(collectible, value))
1335+
removed_data.extend(task.extract_if_all(|key, value| {
1336+
match (key, value) {
1337+
(
1338+
&CachedDataItemKey::InProgressCell { cell },
1339+
CachedDataItemValue::InProgressCell { value },
1340+
) if cell_counters
1341+
.get(&cell.type_id)
1342+
.map_or(true, |start_index| cell.index >= *start_index) =>
1343+
{
1344+
value.event.notify(usize::MAX);
1345+
return true;
13301346
}
1331-
CachedDataItemKey::OutdatedCellDependency { target } => {
1332-
Some(OutdatedEdge::CellDependency(target))
1347+
(&CachedDataItemKey::CellData { cell }, _)
1348+
if cell_counters
1349+
.get(&cell.type_id)
1350+
.map_or(true, |start_index| cell.index >= *start_index) =>
1351+
{
1352+
return true;
1353+
}
1354+
(&CachedDataItemKey::OutdatedChild { task }, _) => {
1355+
old_edges.push(OutdatedEdge::Child(task));
1356+
}
1357+
(
1358+
&CachedDataItemKey::OutdatedCollectible { collectible },
1359+
&CachedDataItemValue::OutdatedCollectible { value },
1360+
) => old_edges.push(OutdatedEdge::Collectible(collectible, value)),
1361+
(&CachedDataItemKey::OutdatedCellDependency { target }, _) => {
1362+
old_edges.push(OutdatedEdge::CellDependency(target));
13331363
}
1334-
CachedDataItemKey::OutdatedOutputDependency { target } => {
1335-
Some(OutdatedEdge::OutputDependency(target))
1364+
(&CachedDataItemKey::OutdatedOutputDependency { target }, _) => {
1365+
old_edges.push(OutdatedEdge::OutputDependency(target));
13361366
}
1337-
CachedDataItemKey::OutdatedCollectiblesDependency { target } => {
1338-
Some(OutdatedEdge::CollectiblesDependency(target))
1367+
(&CachedDataItemKey::OutdatedCollectiblesDependency { target }, _) => {
1368+
old_edges.push(OutdatedEdge::CollectiblesDependency(target));
13391369
}
1340-
CachedDataItemKey::CellDependent { cell, task }
1341-
if removed_cells
1370+
(&CachedDataItemKey::CellDependent { cell, task }, _)
1371+
if cell_counters
13421372
.get(&cell.type_id)
1343-
.map_or(false, |range| range.contains(&cell.index)) =>
1373+
.map_or(true, |start_index| cell.index >= *start_index) =>
13441374
{
1345-
Some(OutdatedEdge::RemovedCellDependent(task, cell.type_id))
1375+
old_edges.push(OutdatedEdge::RemovedCellDependent(task, cell.type_id));
13461376
}
1347-
_ => None,
1348-
})
1349-
.collect::<Vec<_>>()
1377+
_ => {}
1378+
}
1379+
false
1380+
}));
13501381
};
13511382
drop(task);
13521383

‎turbopack/crates/turbo-tasks-backend/src/backend/operation/mod.rs‎

Lines changed: 65 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ use std::{
1111
mem::{take, transmute},
1212
};
1313

14+
use either::Either;
1415
use serde::{Deserialize, Serialize};
1516
use turbo_tasks::{KeyValuePair, SessionId, TaskId, TurboTasksBackendApi};
1617

@@ -370,6 +371,16 @@ pub trait TaskGuard: Debug {
370371
index: CachedDataItemIndex,
371372
) -> impl Iterator<Item = (&CachedDataItemKey, &CachedDataItemValue)>;
372373
fn iter_all(&self) -> impl Iterator<Item = (&CachedDataItemKey, &CachedDataItemValue)>;
374+
fn extract_if<'l, F>(
375+
&'l mut self,
376+
index: CachedDataItemIndex,
377+
f: F,
378+
) -> impl Iterator<Item = CachedDataItem>
379+
where
380+
F: for<'a, 'b> FnMut(&'a CachedDataItemKey, &'b CachedDataItemValue) -> bool + 'l;
381+
fn extract_if_all<'l, F>(&'l mut self, f: F) -> impl Iterator<Item = CachedDataItem>
382+
where
383+
F: for<'a, 'b> FnMut(&'a CachedDataItemKey, &'b CachedDataItemValue) -> bool + 'l;
373384
fn invalidate_serialization(&mut self);
374385
}
375386

@@ -586,6 +597,60 @@ impl<B: BackingStorage> TaskGuard for TaskGuardImpl<'_, B> {
586597
self.task.iter_all()
587598
}
588599

600+
fn extract_if<'l, F>(
601+
&'l mut self,
602+
index: CachedDataItemIndex,
603+
f: F,
604+
) -> impl Iterator<Item = CachedDataItem>
605+
where
606+
F: for<'a, 'b> FnMut(&'a CachedDataItemKey, &'b CachedDataItemValue) -> bool + 'l,
607+
{
608+
if !self.backend.should_persist() || self.task_id.is_transient() {
609+
return Either::Left(self.task.extract_if(Some(index), f));
610+
}
611+
Either::Right(self.task.extract_if(Some(index), f).inspect(|item| {
612+
if item.is_persistent() {
613+
let key = item.key();
614+
let value = item.value();
615+
self.backend
616+
.persisted_storage_log(key.category())
617+
.unwrap()
618+
.lock(self.task_id)
619+
.push(CachedDataUpdate {
620+
key,
621+
task: self.task_id,
622+
value: None,
623+
old_value: Some(value),
624+
});
625+
}
626+
}))
627+
}
628+
629+
fn extract_if_all<'l, F>(&'l mut self, f: F) -> impl Iterator<Item = CachedDataItem>
630+
where
631+
F: for<'a, 'b> FnMut(&'a CachedDataItemKey, &'b CachedDataItemValue) -> bool + 'l,
632+
{
633+
if !self.backend.should_persist() || self.task_id.is_transient() {
634+
return Either::Left(self.task.extract_if_all(f));
635+
}
636+
Either::Right(self.task.extract_if_all(f).inspect(|item| {
637+
if item.is_persistent() {
638+
let key = item.key();
639+
let value = item.value();
640+
self.backend
641+
.persisted_storage_log(key.category())
642+
.unwrap()
643+
.lock(self.task_id)
644+
.push(CachedDataUpdate {
645+
key,
646+
task: self.task_id,
647+
value: None,
648+
old_value: Some(value),
649+
});
650+
}
651+
}))
652+
}
653+
589654
fn invalidate_serialization(&mut self) {
590655
if !self.backend.should_persist() {
591656
return;

0 commit comments

Comments
 (0)