Define ThreadManager::can_finish_session()

anza-xyz · Mar 4, 2025 · aca5764 · aca5764
1 parent 1b35957
commit aca5764
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/unified-scheduler-pool/src/lib.rs b/unified-scheduler-pool/src/lib.rs
@@ -1475,6 +1475,13 @@ impl<S: SpawnableScheduler<TH>, TH: TaskHandler> ThreadManager<S, TH> {
         }
     }
 
+    fn can_finish_session(mode: SchedulingMode, state_machine: &SchedulingStateMachine) -> bool {
+        match mode {
+            BlockVerification => state_machine.has_no_active_task(),
+            BlockProduction => state_machine.has_no_executing_task(),
+        }
+    }
+
     fn take_session_result_with_timings(&mut self) -> ResultWithTimings {
         self.session_result_with_timings.take().unwrap()
     }
@@ -1784,11 +1791,8 @@ impl<S: SpawnableScheduler<TH>, TH: TaskHandler> ThreadManager<S, TH> {
                             },
                         };
 
-                        is_finished = session_ending
-                            && (scheduling_mode == BlockVerification
-                                && state_machine.has_no_active_task()
-                                || scheduling_mode == BlockProduction
-                                    && state_machine.has_no_executing_task());
+                        is_finished =
+                            session_ending && Self::can_finish_session(scheduling_mode, &state_machine);
                     }
                     assert!(mem::replace(&mut is_finished, false));