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
mod kill_switch;
pub mod manager;
mod name;
pub use self::{manager::Manager, name::Name};
use self::kill_switch::KillSwitch;
use crate::{FrameworkError, FrameworkErrorKind::ThreadError};
use std::{io, sync::Arc, thread};
pub type JoinHandle = thread::JoinHandle<()>;
#[derive(Debug)]
pub struct Thread {
name: Name,
kill_switch: Arc<KillSwitch>,
handle: JoinHandle,
}
impl Thread {
pub fn spawn<F>(name: Name, f: F) -> Result<Self, FrameworkError>
where
F: FnOnce() + Send + 'static,
{
let kill_switch = Arc::new(KillSwitch::new());
let handle = spawn_thread(name.clone(), Arc::clone(&kill_switch), f)?;
Ok(Self {
name,
kill_switch,
handle,
})
}
pub fn name(&self) -> &Name {
&self.name
}
pub fn request_termination(&self) {
self.kill_switch.throw();
}
pub fn join(self) -> Result<(), FrameworkError> {
self.request_termination();
self.handle
.join()
.map_err(|e| format_err!(ThreadError, "{:?}", e))?;
Ok(())
}
}
pub fn should_terminate() -> bool {
kill_switch::is_thrown()
}
fn spawn_thread<F>(name: Name, kill_switch: Arc<KillSwitch>, f: F) -> Result<JoinHandle, io::Error>
where
F: FnOnce() + Send + 'static,
{
thread::Builder::new().name(name.into()).spawn(move || {
kill_switch::set(kill_switch);
f()
})
}