Robin Mueller
0a21fcf23a
Some checks failed
Rust/sat-rs/pipeline/pr-main There was a failure building this commit
39 lines
1.4 KiB
Rust
39 lines
1.4 KiB
Rust
use satrs_core::pool::{
|
|
PoolGuard, PoolProviderMemInPlace, StaticMemoryPool, StaticPoolConfig, StoreAddr,
|
|
};
|
|
use std::ops::DerefMut;
|
|
use std::sync::mpsc;
|
|
use std::sync::mpsc::{Receiver, Sender};
|
|
use std::sync::{Arc, RwLock};
|
|
use std::thread;
|
|
|
|
const DUMMY_DATA: [u8; 4] = [0, 1, 2, 3];
|
|
|
|
#[test]
|
|
fn threaded_usage() {
|
|
let pool_cfg = StaticPoolConfig::new(vec![(16, 6), (32, 3), (8, 12)]);
|
|
let shared_pool = Arc::new(RwLock::new(StaticMemoryPool::new(pool_cfg)));
|
|
let shared_clone = shared_pool.clone();
|
|
let (tx, rx): (Sender<StoreAddr>, Receiver<StoreAddr>) = mpsc::channel();
|
|
let jh0 = thread::spawn(move || {
|
|
let mut dummy = shared_pool.write().unwrap();
|
|
let addr = dummy.add(&DUMMY_DATA).expect("Writing data failed");
|
|
tx.send(addr).expect("Sending store address failed");
|
|
});
|
|
|
|
let jh1 = thread::spawn(move || {
|
|
let addr;
|
|
{
|
|
addr = rx.recv().expect("Receiving store address failed");
|
|
let mut pool_access = shared_clone.write().unwrap();
|
|
let pg = PoolGuard::new(pool_access.deref_mut(), addr);
|
|
let read_res = pg.read().expect("Reading failed");
|
|
assert_eq!(read_res, DUMMY_DATA);
|
|
}
|
|
let pool_access = shared_clone.read().unwrap();
|
|
assert!(!pool_access.has_element_at(&addr).expect("Invalid address"));
|
|
});
|
|
jh0.join().unwrap();
|
|
jh1.join().unwrap();
|
|
}
|