fix: optimized node runners calculation

This commit is contained in:
Marvin Zhang
2024-12-11 20:43:40 +08:00
parent 6b78afe95c
commit 1fe74fa8a5
4 changed files with 26 additions and 27 deletions

View File

@@ -215,13 +215,13 @@ func (svc *Service) getRunnerCount() (count int) {
if svc.cfgSvc.IsMaster() {
count, err = service.NewModelService[models.Task]().Count(query)
if err != nil {
trace.PrintError(err)
log.Errorf("failed to count tasks: %v", err)
return
}
} else {
count, err = client.NewModelService[models.Task]().Count(query)
if err != nil {
trace.PrintError(err)
log.Errorf("failed to count tasks: %v", err)
return
}
}
@@ -260,11 +260,8 @@ func (svc *Service) updateNodeStatus() (err error) {
return err
}
// available runners of handler
ar := n.MaxRunners - svc.getRunnerCount()
// set available runners
n.AvailableRunners = ar
n.CurrentRunners = svc.getRunnerCount()
// save node
n.SetUpdated(n.CreatedBy)