berkus
/
actix-net
mirror of https://github.com/fafhrd91/actix-net
1
0
Fork 0
Code Issues Releases Wiki Activity

Refine runtime match

This commit is contained in:
fakeshadow 2021-04-17 20:29:48 +08:00
parent d428ac275e
commit 297b172644
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
actix-server/src/worker.rs
View File

@ -272,11 +272,6 @@ impl ServerWorker {
thread::Builder::new()
.name(format!("actix-server-worker-{}", idx))
.spawn(move || {
// Conditionally setup actix system.
if let Some(system) = system {
System::set_current(system);
}
// Prepare service construct future.
let fut = async {
let mut services = Vec::new();
@ -301,8 +296,8 @@ impl ServerWorker {
// All future runs in a LocalSet for being able to run !Send future.
let local = tokio::task::LocalSet::new();
match rt_handle {
Ok(handle) => {
match (rt_handle, system) {
(Ok(handle), None) => {
// Use existing tokio runtime with handle.
let res = handle.block_on(local.run_until(fut));
@ -330,7 +325,12 @@ impl ServerWorker {
Err(e) => f(Some(e)),
}
}
Err(_) => {
(_, system) => {
// Conditionally setup actix system.
if let Some(system) = system {
System::set_current(system);
}
// No existing tokio runtime found. Start new runtime.
let rt = tokio::runtime::Builder::new_current_thread()
.enable_all()