spdf/elektron
Archived
1
0
Fork 0
Code Issues 4 Pull requests 1 Projects Releases Packages Wiki Activity

Handle case: host -> slaveID may not be available

Browse source
This commit is contained in:
Akash Kothawale 2018-02-05 02:25:32 -05:00 • committed by Pradyumna Kaushik
parent 3d5bb772fb
commit c52bfb08c5
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
pcp/pcp.go
View file

@ -11,6 +11,8 @@ import (
"syscall"
"time"
"github.com/montanaflynn/stats"
"github.com/mesos/mesos-go/api/v0/mesosproto"
"path/filepath"
)
func Start(quit chan struct{}, logging *bool, logMType chan elecLogDef.LogMessageType, logMsg chan string, s scheduler.Scheduler) {
@ -61,11 +63,12 @@ func Start(quit chan struct{}, logging *bool, logMType chan elecLogDef.LogMessag
for i := 0; i < 8; i++ {
host := fmt.Sprintf("stratos-00%d.cs.binghamton.edu", i+1)
slaveID := baseSchedRef.HostNameToSlaveID[host]
tasksRunning := len(baseSchedRef.Running[slaveID])
if tasksRunning > 0 {
cpuTaskShares[i] = cpuUtils[i] / float64(tasksRunning)
memTaskShares[i] = memUtils[i] / float64(tasksRunning)
if slaveID, ok := baseSchedRef.HostNameToSlaveID[host]; ok {
tasksRunning := len(baseSchedRef.Running[slaveID])
if tasksRunning > 0 {
cpuTaskShares[i] = cpuUtils[i] / float64(tasksRunning)
memTaskShares[i] = memUtils[i] / float64(tasksRunning)
}
}
}