proxmox-mirrors/proxmox
1
0
Fork 1
mirror of https://git.proxmox.com/git/proxmox synced 2025-08-08 07:48:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

rustfmt

Browse Source 
Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
This commit is contained in:
Wolfgang Bumiller 2022-01-20 10:12:02 +01:00
parent 175648763d
commit ff132e93c6
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
proxmox-async/src/runtime.rs
View File

@ -2,7 +2,7 @@
use std::cell::RefCell; use std::cell::RefCell;
use std::future::Future; use std::future::Future;
use std::sync::{Arc, Weak, Mutex}; use std::sync::{Arc, Mutex, Weak};
use std::task::{Context, Poll, Waker}; use std::task::{Context, Poll, Waker};
use std::thread::{self, Thread}; use std::thread::{self, Thread};
@ -15,8 +15,7 @@ thread_local! {
} }
fn is_in_tokio() -> bool { fn is_in_tokio() -> bool {
tokio::runtime::Handle::try_current() tokio::runtime::Handle::try_current().is_ok()
.is_ok()
} }
fn is_blocking() -> bool { fn is_blocking() -> bool {
@ -59,16 +58,19 @@ extern "C" {
/// 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_with_builder<F: Fn() -> runtime::Builder>(get_builder: F) -> 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 mut builder = get_builder(); let mut builder = get_builder();
builder.on_thread_stop(|| { builder.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();
}
}); });
let runtime = builder.build().expect("failed to spawn tokio runtime"); let runtime = builder.build().expect("failed to spawn tokio runtime");
@ -83,7 +85,6 @@ pub fn get_runtime_with_builder<F: Fn() -> runtime::Builder>(get_builder: F) ->
/// ///
/// This calls get_runtime_with_builder() using the tokio default threaded scheduler /// This calls get_runtime_with_builder() using the tokio default threaded scheduler
pub fn get_runtime() -> Arc<Runtime> { pub fn get_runtime() -> Arc<Runtime> {
get_runtime_with_builder(|| { get_runtime_with_builder(|| {
let mut builder = runtime::Builder::new_multi_thread(); let mut builder = runtime::Builder::new_multi_thread();
builder.enable_all(); builder.enable_all();
@ -91,7 +92,6 @@ pub fn get_runtime() -> Arc<Runtime> {
}) })
} }
/// 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)