-
Notifications
You must be signed in to change notification settings - Fork 14
/
latency.rs
123 lines (102 loc) · 2.81 KB
/
latency.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
//! Latency benchmarks
#![feature(test)]
extern crate mioco;
extern crate test;
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use test::Bencher;
struct SendBencher(*mut Bencher);
// Don't judge me. -- dpc
unsafe impl Send for SendBencher {}
//
// WARNING!
//
// Some pieces here are very fragile and will short-circuit on
// any spurious wakeup, eg. when previous task leaves wakeup
// call that was not received by `wait`.
//
#[bench]
fn mpsc_pingpong(b: &mut Bencher) {
let (tx1, rx1) = mioco::sync::mpsc::channel();
let (tx2, rx2) = mioco::sync::mpsc::channel();
let b = SendBencher(b as *mut Bencher);
let join1 = mioco::spawn(move || {
let mut prev = 0;
loop {
tx2.send(prev + 1).unwrap();
match rx1.recv() {
Ok(x) => prev = x,
Err(_) => break,
}
}
});
let join2 = mioco::spawn(move || {
let b = unsafe { &mut *b.0 };
b.iter(|| {
let x = rx2.recv().unwrap();
tx1.send(x + 1).unwrap();
});
});
join1.join().unwrap();
join2.join().unwrap();
}
#[bench]
fn notify_pingpong(b: &mut Bencher) {
let (tx1, rx1) = mioco::sync::notify::channel();
let (tx2, rx2) = mioco::sync::notify::channel();
let b = SendBencher(b as *mut Bencher);
let finished = Arc::new(AtomicBool::new(false));
let join1 = mioco::spawn({
let finished = finished.clone();
move || {
while !finished.load(Ordering::SeqCst) {
rx1.reset();
tx2.notify();
rx1.wait();
}
}
});
let join2 = mioco::spawn(move || {
let b = unsafe { &mut *b.0 };
b.iter(|| {
rx2.wait();
rx2.reset();
tx1.notify();
});
rx2.wait();
finished.store(true, Ordering::SeqCst);
tx1.notify();
});
join1.join().unwrap();
join2.join().unwrap();
}
#[bench]
fn broadcast_pingpong(b: &mut Bencher) {
let (tx1, rx1) = mioco::sync::broadcast::channel();
let (tx2, rx2) = mioco::sync::broadcast::channel();
let b = SendBencher(b as *mut Bencher);
let finished = Arc::new(AtomicBool::new(false));
let join1 = mioco::spawn({
let finished = finished.clone();
move || {
while !finished.load(Ordering::SeqCst) {
rx1.reset();
tx2.notify();
rx1.wait();
}
}
});
let join2 = mioco::spawn(move || {
let b = unsafe { &mut *b.0 };
b.iter(|| {
rx2.wait();
rx2.reset();
tx1.notify();
});
rx2.wait();
finished.store(true, Ordering::SeqCst);
tx1.notify();
});
join1.join().unwrap();
join2.join().unwrap();
}