milksad
/
rust-bitcoin-unsafe-fast
14
0
Fork 0
Code Issues Pull Requests Activity

utxoset: Fix parallel script checking to use only as many threads as CPUs

This commit is contained in:
Andrew Poelstra 2014-08-14 17:14:32 -07:00
parent fc04462682
commit c28c261b74
1 changed files with 36 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/blockdata/utxoset.rs
View File

@ -18,8 +18,10 @@
//! index of UTXOs. //! index of UTXOs.
//! //!
use std::cmp;
use std::collections::HashMap; use std::collections::HashMap;
use std::mem; use std::mem;
use std::os::num_cpus;
use std::sync::Future; use std::sync::Future;
use blockdata::transaction::{Transaction, TxOut}; use blockdata::transaction::{Transaction, TxOut};
@ -207,11 +209,17 @@ impl UtxoSet {
let mut future_vec = Vec::with_capacity(block.txdata.len() - 1); let mut future_vec = Vec::with_capacity(block.txdata.len() - 1);
// skip the genesis since we don't validate this script. (TODO this might // skip the genesis since we don't validate this script. (TODO this might
// be a consensus bug since we don't even check that the opcodes make sense.) // be a consensus bug since we don't even check that the opcodes make sense.)
for tx in block.txdata.iter().skip(1) { let n_threads = cmp::min(block.txdata.len() - 1, num_cpus());
for j in range(0, n_threads) {
let n_elems = block.txdata.len() - 1;
let start = 1 + j * n_elems / n_threads;
let end = cmp::min(n_elems, 1 + (j + 1) * n_elems / n_threads);
let s = self as *mut _ as *const UtxoSet; let s = self as *mut _ as *const UtxoSet;
let tx = tx as *const _; let txes = &block.txdata as *const _;
future_vec.push(Future::spawn(proc() { future_vec.push(Future::spawn(proc() {
let tx = unsafe {&*tx}; let txes = unsafe {&*txes};
for tx in txes.slice(start, end).iter() {
for (n, input) in tx.input.iter().enumerate() { for (n, input) in tx.input.iter().enumerate() {
let txo = unsafe { (*s).get_utxo(input.prev_hash, input.prev_index) }; let txo = unsafe { (*s).get_utxo(input.prev_hash, input.prev_index) };
match txo { match txo {
@ -241,6 +249,7 @@ impl UtxoSet {
None => { return false; } None => { return false; }
} }
} }
}
true true
})); }));
} }