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

Added proactive cluster wide capping with ranked tasks as another scheduler.

Browse source
This commit is contained in:
Pradyumna Kaushik 2016-11-25 18:00:55 -05:00 • committed by Renan DelValle
parent 892d9b86dc
commit f67773dcde
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
schedulers/proactiveclusterwidecappingranked.go
View file

@ -38,11 +38,11 @@ func (_ *ProactiveClusterwideCapRanked) takeOffer(offer *mesos.Offer, task def.T
// electronScheduler implements the Scheduler interface // electronScheduler implements the Scheduler interface
type ProactiveClusterwideCapRanked struct { type ProactiveClusterwideCapRanked struct {
tasksCreated int tasksCreated int
tasksRunning int tasksRunning int
tasks []def.Task tasks []def.Task
metrics map[string]def.Metric metrics map[string]def.Metric
running map[string]map[string]bool running map[string]map[string]bool
taskMonitor map[string][]def.Task // store tasks that are currently running. taskMonitor map[string][]def.Task // store tasks that are currently running.
availablePower map[string]float64 // available power for each node in the cluster. availablePower map[string]float64 // available power for each node in the cluster.
totalPower map[string]float64 // total power for each node in the cluster. totalPower map[string]float64 // total power for each node in the cluster.
@ -86,7 +86,7 @@ func NewProactiveClusterwideCapRanked(tasks []def.Task, ignoreWatts bool) *Proac
ticker: time.NewTicker(10 * time.Second), ticker: time.NewTicker(10 * time.Second),
recapTicker: time.NewTicker(20 * time.Second), recapTicker: time.NewTicker(20 * time.Second),
isCapping: false, isCapping: false,
isRecapping: false, isRecapping: false,
} }
return s return s
} }
@ -114,10 +114,10 @@ func (s *ProactiveClusterwideCapRanked) newTask(offer *mesos.Offer, task def.Tas
task.SetTaskID(*proto.String("electron-" + taskName)) task.SetTaskID(*proto.String("electron-" + taskName))
// Add task to the list of tasks running on the node. // Add task to the list of tasks running on the node.
s.running[offer.GetSlaveId().GoString()][taskName] = true s.running[offer.GetSlaveId().GoString()][taskName] = true
if len(s.taskMonitor[offer.GetSlaveId().GoString()]) == 0 { if len(s.taskMonitor[*offer.Hostname]) == 0 {
s.taskMonitor[offer.GetSlaveId().GoString()] = []def.Task{task} s.taskMonitor[*offer.Hostname] = []def.Task{task}
} else { } else {
s.taskMonitor[offer.GetSlaveId().GoString()] = append(s.taskMonitor[offer.GetSlaveId().GoString()], task) s.taskMonitor[*offer.Hostname] = append(s.taskMonitor[*offer.Hostname], task)
} }
resources := []*mesos.Resource{ resources := []*mesos.Resource{