Skip to content

Commit

Permalink
Ensure reinitialize() is maintained for new fields
Browse files Browse the repository at this point in the history
  • Loading branch information
ryoqun committed Mar 4, 2024
1 parent 32ce932 commit 8b317eb
Showing 1 changed file with 17 additions and 5 deletions.
22 changes: 17 additions & 5 deletions unified-scheduler-logic/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -738,11 +738,23 @@ impl SchedulingStateMachine {
pub fn reinitialize(&mut self) {
assert!(self.has_no_active_task());
assert_eq!(self.unblocked_task_queue.len(), 0);
self.last_task_index = None;
self.active_task_count.reset_to_zero();
self.handled_task_count.reset_to_zero();
self.unblocked_task_count.reset_to_zero();
self.total_task_count.reset_to_zero();
// nice trick to ensure all fields are handled here if new one is added.
let Self {
last_task_index,
unblocked_task_queue: _,
active_task_count,
handled_task_count,
unblocked_task_count,
total_task_count,
count_token: _,
usage_queue_token: _,
// don't add ".." here
} = self;
*last_task_index = None;
active_task_count.reset_to_zero();
handled_task_count.reset_to_zero();
unblocked_task_count.reset_to_zero();
total_task_count.reset_to_zero();
}

/// Creates a new instance of [`SchedulingStateMachine`] with its `unsafe` fields created as
Expand Down

0 comments on commit 8b317eb

Please sign in to comment.