respect time budget in main GC loop

rerun-io · Nov 30, 2023 · 2d00c22 · 2d00c22
1 parent 2ab2da9
commit 2d00c22
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
@@ -249,6 +249,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;
@@ -291,7 +292,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;
             }