mirrors/ntex
1
0
Fork 0
mirror of https://github.com/ntex-rs/ntex.git synced 2025-04-03 21:07:39 +03:00
Code Issues Projects Releases Packages Wiki Activity

Refactor worker management (#312)

Browse source
This commit is contained in:
Nikolay Kim 2024-03-23 07:17:04 +01:00 • committed by GitHub
parent ca16e4a5e4
commit d393d87164
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
46 changed files with 1925 additions and 1782 deletions
Download patch file Download diff file Expand all files Collapse all files

16
ntex-rt/src/arbiter.rs
View file

@ -1,7 +1,7 @@
#![allow(clippy::let_underscore_future)]
use std::any::{Any, TypeId};
use std::sync::atomic::{AtomicUsize, Ordering};
use std::task::{Context, Poll};
use std::task::{ready, Context, Poll};
use std::{cell::RefCell, collections::HashMap, fmt, future::Future, pin::Pin, thread};
use async_channel::{unbounded, Receiver, Sender};
@ -320,10 +320,17 @@ impl Future for SystemArbiter {
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
loop {
match Pin::new(&mut self.commands).poll_next(cx) {
Poll::Ready(None) => return Poll::Ready(()),
Poll::Ready(Some(cmd)) => match cmd {
let cmd = ready!(Pin::new(&mut self.commands).poll_next(cx));
log::debug!("Received system command: {:?}", cmd);
match cmd {
None => {
log::debug!("System stopped");
return Poll::Ready(());
}
Some(cmd) => match cmd {
SystemCommand::Exit(code) => {
log::debug!("Stopping system with {} code", code);
// stop arbiters
for arb in self.arbiters.values() {
arb.stop();
@ -340,7 +347,6 @@ impl Future for SystemArbiter {
self.arbiters.remove(&name);
}
},
Poll::Pending => return Poll::Pending,
}
}
}