52 lines
1.7 KiB
Rust
52 lines
1.7 KiB
Rust
use solana_ledger::entry;
|
|
use solana_ledger::{
|
|
blockstore::{self, Blockstore},
|
|
get_tmp_ledger_path,
|
|
};
|
|
use solana_sdk::hash::Hash;
|
|
use std::sync::Arc;
|
|
use std::thread::Builder;
|
|
|
|
#[test]
|
|
fn test_multiple_threads_insert_shred() {
|
|
let blockstore_path = get_tmp_ledger_path!();
|
|
let blockstore = Arc::new(Blockstore::open(&blockstore_path).unwrap());
|
|
|
|
for _ in 0..100 {
|
|
let num_threads = 10;
|
|
|
|
// Create `num_threads` different ticks in slots 1..num_threads + 1, all
|
|
// with parent = slot 0
|
|
let threads: Vec<_> = (0..num_threads)
|
|
.map(|i| {
|
|
let entries = entry::create_ticks(1, 0, Hash::default());
|
|
let shreds = blockstore::entries_to_test_shreds(entries, i + 1, 0, false, 0);
|
|
let blockstore_ = blockstore.clone();
|
|
Builder::new()
|
|
.name("blockstore-writer".to_string())
|
|
.spawn(move || {
|
|
blockstore_.insert_shreds(shreds, None, false).unwrap();
|
|
})
|
|
.unwrap()
|
|
})
|
|
.collect();
|
|
|
|
for t in threads {
|
|
t.join().unwrap()
|
|
}
|
|
|
|
// Check slot 0 has the correct children
|
|
let mut meta0 = blockstore.meta(0).unwrap().unwrap();
|
|
meta0.next_slots.sort_unstable();
|
|
let expected_next_slots: Vec<_> = (1..num_threads + 1).collect();
|
|
assert_eq!(meta0.next_slots, expected_next_slots);
|
|
|
|
// Delete slots for next iteration
|
|
blockstore.purge_and_compact_slots(0, num_threads + 1);
|
|
}
|
|
|
|
// Cleanup
|
|
drop(blockstore);
|
|
Blockstore::destroy(&blockstore_path).expect("Expected successful database destruction");
|
|
}
|