Skip to content

Commit f05a1df

Browse files
committed
check task finished before timeout
1 parent ba952b0 commit f05a1df

File tree

1 file changed

+3
-0
lines changed
  • interactive_engine/executor/engine/pegasus/pegasus/src

1 file changed

+3
-0
lines changed

interactive_engine/executor/engine/pegasus/pegasus/src/worker.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,9 @@ impl<D: Data, T: Debug + Send + 'static> Task for Worker<D, T> {
293293

294294
fn check_ready(&mut self) -> TaskState {
295295
let _g = crate::worker_id::guard(self.id);
296+
if self.is_finished && self.peer_guard.load(Ordering::SeqCst) == 0 {
297+
return TaskState::Finished;
298+
}
296299
if self.check_cancel() {
297300
self.sink.set_cancel_hook(true);
298301
return TaskState::Finished;

0 commit comments

Comments
 (0)