tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

src/tools/runtime.rs: implement get_runtime_with_builder

Browse Source
This commit is contained in:
Dietmar Maurer 2020-07-07 10:11:04 +02:00
parent 1ff840ffad
commit f91d58e157
1 changed files with 24 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/tools/runtime.rs
View File

@ -56,30 +56,41 @@ extern {
/// ///
/// This makes sure that tokio's worker threads are marked for us so that we know whether we /// This makes sure that tokio's worker threads are marked for us so that we know whether we
/// can/need to use `block_in_place` in our `block_on` helper. /// can/need to use `block_in_place` in our `block_on` helper.
pub fn get_runtime() -> Arc<Runtime> { pub fn get_runtime_with_builder<F: Fn() -> runtime::Builder>(get_builder: F) -> Arc<Runtime> {
let mut guard = RUNTIME.lock().unwrap(); let mut guard = RUNTIME.lock().unwrap();
if let Some(rt) = guard.upgrade() { return rt; } if let Some(rt) = guard.upgrade() { return rt; }
let rt = Arc::new( let mut builder = get_builder();
runtime::Builder::new() builder.on_thread_stop(|| {
.on_thread_stop(|| {
// avoid openssl bug: https://github.com/openssl/openssl/issues/6214 // avoid openssl bug: https://github.com/openssl/openssl/issues/6214
// call OPENSSL_thread_stop to avoid race with openssl cleanup handlers // call OPENSSL_thread_stop to avoid race with openssl cleanup handlers
unsafe { OPENSSL_thread_stop(); } unsafe { OPENSSL_thread_stop(); }
}) });
.threaded_scheduler()
.enable_all() let runtime = builder.build().expect("failed to spawn tokio runtime");
.build() let rt = Arc::new(runtime);
.expect("failed to spawn tokio runtime")
);
*guard = Arc::downgrade(&rt.clone()); *guard = Arc::downgrade(&rt.clone());
rt rt
} }
/// Get or create the current main tokio runtime.
///
/// This calls get_runtime_with_builder() using the tokio default threaded scheduler
pub fn get_runtime() -> Arc<Runtime> {
get_runtime_with_builder(|| {
let mut builder = runtime::Builder::new();
builder.threaded_scheduler();
builder.enable_all();
builder
})
}
/// Block on a synchronous piece of code. /// Block on a synchronous piece of code.
pub fn block_in_place<R>(fut: impl FnOnce() -> R) -> R { pub fn block_in_place<R>(fut: impl FnOnce() -> R) -> R {
// don't double-exit the context (tokio doesn't like that) // don't double-exit the context (tokio doesn't like that)