respect time budget in main GC loop

rerun-io · Dec 2, 2023 · e3d6267 · e3d6267
1 parent 0f24fa4
commit e3d6267
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/re_arrow_store/src/store_gc.rs b/crates/re_arrow_store/src/store_gc.rs
@@ -248,6 +248,7 @@ impl DataStore {
             ..
         } = self;
 
+        let now = std::time::Instant::now();
         for (&row_id, (timepoint, entity_path_hash)) in &metadata_registry.registry {
             if protected_rows.contains(&row_id) {
                 batch_is_protected = true;
@@ -290,7 +291,7 @@ impl DataStore {
                 diffs.push(dropped);
             }
 
-            if num_bytes_to_drop <= 0.0 {
+            if now.elapsed() >= options.time_budget || num_bytes_to_drop <= 0.0 {
                 break;
             }