Merge pull request #68 from mempool/junderw/threadpool-builder-fix

ThreadPoolBuilder can fail when resources are busy
mempool · Feb 23, 2024 · f78c07a · f78c07a
2 parents 13e5023 + 1cb2715
commit f78c07a
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/src/new_index/schema.rs b/src/new_index/schema.rs
@@ -1095,11 +1095,23 @@ fn lookup_txos(
     outpoints: &BTreeSet<OutPoint>,
     allow_missing: bool,
 ) -> HashMap<OutPoint, TxOut> {
-    let pool = rayon::ThreadPoolBuilder::new()
-        .num_threads(16) // we need to saturate SSD IOPS
-        .thread_name(|i| format!("lookup-txo-{}", i))
-        .build()
-        .unwrap();
+    let mut loop_count = 10;
+    let pool = loop {
+        match rayon::ThreadPoolBuilder::new()
+            .num_threads(16) // we need to saturate SSD IOPS
+            .thread_name(|i| format!("lookup-txo-{}", i))
+            .build()
+        {
+            Ok(pool) => break pool,
+            Err(e) => {
+                if loop_count == 0 {
+                    panic!("schema::lookup_txos failed to create a ThreadPool: {}", e);
+                }
+                std::thread::sleep(std::time::Duration::from_millis(50));
+                loop_count -= 1;
+            }
+        }
+    };
     pool.install(|| {
         outpoints
             .par_iter()