From c52bfb08c548f87fe5935203628a1d8d146d6bed Mon Sep 17 00:00:00 2001 From: Akash Kothawale Date: Mon, 5 Feb 2018 02:25:32 -0500 Subject: [PATCH] Handle case: host -> slaveID may not be available --- pcp/pcp.go | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/pcp/pcp.go b/pcp/pcp.go index 6dabcb2..c5e3e04 100644 --- a/pcp/pcp.go +++ b/pcp/pcp.go @@ -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) + } } }