Skip to content

Commit cb357fb

Browse files
committed
gang: track incomplete_tasks on-demand via bind/unbind hooks
1 parent f89eeca commit cb357fb

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

  • session_manager/src/scheduler/plugins

session_manager/src/scheduler/plugins/gang.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,7 @@ impl Plugin for GangPlugin {
154154
fn on_session_bind(&mut self, ssn: SessionInfoPtr) {
155155
if let Some(state) = self.ssn_state.get_mut(&ssn.id) {
156156
state.bound += 1;
157+
state.incomplete_tasks = state.incomplete_tasks.saturating_sub(1);
157158
}
158159
}
159160

@@ -166,6 +167,7 @@ impl Plugin for GangPlugin {
166167
fn on_session_unbind(&mut self, ssn: SessionInfoPtr) {
167168
if let Some(state) = self.ssn_state.get_mut(&ssn.id) {
168169
state.bound = state.bound.saturating_sub(1);
170+
state.incomplete_tasks = state.incomplete_tasks.saturating_add(1);
169171
}
170172
}
171173
}

0 commit comments

Comments
 (0)