actix/actix-net
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-net synced 2024-11-27 22:32:57 +01:00
Code Issues Releases Wiki Activity

fix switching to unavailable state for worker

Browse Source
This commit is contained in:
Nikolay Kim 2018-09-14 00:07:50 -07:00
parent ec7757f032
commit 9888c1c5e6
1 changed files with 31 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/server/worker.rs
View File

@ -210,6 +210,7 @@ impl Future for Worker {
WorkerState::Unavailable(mut conns) => { WorkerState::Unavailable(mut conns) => {
match self.check_readiness() { match self.check_readiness() {
Ok(true) => { Ok(true) => {
trace!("Serveice is available");
self.state = WorkerState::Available; self.state = WorkerState::Available;
// process requests from wait queue // process requests from wait queue
@ -227,10 +228,12 @@ impl Future for Worker {
) )
} }
Ok(false) => { Ok(false) => {
trace!("Serveice is unavailable");
self.state = WorkerState::Unavailable(conns); self.state = WorkerState::Unavailable(conns);
return self.poll(); return self.poll();
} }
Err(idx) => { Err(idx) => {
trace!("Serveice failed, restarting");
self.state = WorkerState::Restarting( self.state = WorkerState::Restarting(
idx, idx,
self.factories[idx].create(), self.factories[idx].create(),
@ -247,6 +250,7 @@ impl Future for Worker {
return Ok(Async::NotReady); return Ok(Async::NotReady);
} }
Err(idx) => { Err(idx) => {
trace!("Serveice failed, restarting");
self.state = WorkerState::Restarting(idx, self.factories[idx].create()); self.state = WorkerState::Restarting(idx, self.factories[idx].create());
return self.poll(); return self.poll();
} }
@ -303,8 +307,8 @@ impl Future for Worker {
loop { loop {
match self.rx.poll() { match self.rx.poll() {
// handle incoming tcp stream // handle incoming tcp stream
Ok(Async::Ready(Some(WorkerCommand::Message(msg)))) => match self Ok(Async::Ready(Some(WorkerCommand::Message(msg)))) => {
.check_readiness() match self.check_readiness()
{ {
Ok(true) => { Ok(true) => {
let guard = self.conns.get(); let guard = self.conns.get();
@ -316,16 +320,21 @@ impl Future for Worker {
val val
}), }),
); );
continue
} }
Ok(false) => { Ok(false) => {
trace!("Serveice is unsavailable");
self.availability.set(false); self.availability.set(false);
self.state = WorkerState::Unavailable(vec![msg]); self.state = WorkerState::Unavailable(vec![msg]);
} }
Err(idx) => { Err(idx) => {
trace!("Serveice failed, restarting");
self.availability.set(false); self.availability.set(false);
self.state = self.state =
WorkerState::Restarting(idx, self.factories[idx].create()); WorkerState::Restarting(idx, self.factories[idx].create());
} }
}
return self.poll();
}, },
// `StopWorker` message handler // `StopWorker` message handler
Ok(Async::Ready(Some(WorkerCommand::Stop(graceful, tx)))) => { Ok(Async::Ready(Some(WorkerCommand::Stop(graceful, tx)))) => {