YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-12-12 23:36:44 +00:00
Code Issues Releases Wiki Activity

update shceduler_fair

Browse Source
This commit is contained in:
Newnius
2019-08-01 14:00:24 +08:00
parent ba630f0318
commit f5ef9b00a6
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/scheduler_fair.go
View File

@@ -21,7 +21,7 @@ type SchedulerFair struct {
scheduling sync.Mutex
jobs map[string]*JobManager
nextQueue string
resourceAllocations map[string]ResourceCount
resourceAllocations map[string]*ResourceCount
}
type FairJobSorter []Job
@@ -45,7 +45,7 @@ func (scheduler *SchedulerFair) Start() {
scheduler.nextQueue = "default"
scheduler.queues = map[string][]Job{}
scheduler.queues["default"] = []Job{}
scheduler.resourceAllocations = map[string]ResourceCount{}
scheduler.resourceAllocations = map[string]*ResourceCount{}
go func() {
for {
@@ -182,9 +182,9 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task) NodeStatus {
}
go func(res NodeStatus) {
if _, ok := scheduler.resourceAllocations[job.Group]; !ok {
scheduler.resourceAllocations[job.Group] = ResourceCount{}
scheduler.resourceAllocations[job.Group] = &ResourceCount{}
}
cnt, _ := scheduler.resourceAllocations[job.Group]
cnt, _ :=scheduler.resourceAllocations[job.Group]
cnt.CPU += res.MemTotal
cnt.Memory += res.NumCPU
for _, v := range res.Status {
@@ -210,7 +210,7 @@ func (scheduler *SchedulerFair) ReleaseResource(job Job, agent NodeStatus) {
}
go func(res NodeStatus) {
if _, ok := scheduler.resourceAllocations[job.Group]; !ok {
scheduler.resourceAllocations[job.Group] = ResourceCount{}
scheduler.resourceAllocations[job.Group] = &ResourceCount{}
}
cnt, _ := scheduler.resourceAllocations[job.Group]
cnt.CPU -= res.MemTotal