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
#![cfg(feature = "v1")]
use tokio1 as tokio;
use tokio::runtime::{Builder, Runtime};
use tokio::task::JoinHandle;
use std::future::Future;
use std::sync::{Mutex, RwLock, RwLockReadGuard};
struct Common {
used_runtime: bool,
runtime: Option<Runtime>,
}
lazy_static::lazy_static! {
static ref COMMON : Mutex<Common> = Mutex::new(Common{
used_runtime: false,
runtime: None,
});
static ref RUNTIME : RwLock<Runtime> = {
let mut c = COMMON.lock().expect("unable to lock untokio::v1::COMMON");
c.used_runtime = true;
RwLock::new(c.runtime.take().unwrap_or_else(|| Builder::new_multi_thread()
.enable_all()
.thread_name("untokio::v1")
.build()
.expect("unable to create untokio::v1::RUNTIME")
))
};
}
pub fn runtime() -> RwLockReadGuard<'static, Runtime> {
RUNTIME.read().expect("unable to lock untokio::v1::RUNTIME")
}
pub fn try_set_runtime(runtime: Runtime) -> Result<(), &'static str> {
let mut c = COMMON.lock().map_err(|_| "unable to lock untokio::v1::COMMON")?;
if c.used_runtime { return Err("untokio::v1::RUNTIME already in use"); }
c.runtime = Some(runtime);
Ok(())
}
pub fn set_runtime(runtime: Runtime) {
try_set_runtime(runtime).unwrap();
}
pub fn spawn<F>(future: F) -> JoinHandle<F::Output> where F : Future + Send + 'static, F::Output : Send + 'static {
runtime().spawn(future)
}