sched: Make baseScheduler publicly accessible

This commit is contained in:
Akash Kothawale 2018-01-30 14:12:37 -05:00 committed by Pradyumna Kaushik
parent 9599588fb5
commit 6f0f3788b9
7 changed files with 57 additions and 58 deletions

View file

@ -14,7 +14,7 @@ import (
// Decides if to take an offer or not
func (s *BinPackSortedWatts) takeOffer(spc SchedPolicyContext, offer *mesos.Offer, task def.Task, totalCPU, totalRAM, totalWatts float64) bool {
baseSchedRef := spc.(*baseScheduler)
baseSchedRef := spc.(*BaseScheduler)
cpus, mem, watts := offerUtils.OfferAgg(offer)
//TODO: Insert watts calculation here instead of taking them as a parameter
@ -37,7 +37,7 @@ type BinPackSortedWatts struct {
func (s *BinPackSortedWatts) ConsumeOffers(spc SchedPolicyContext, driver sched.SchedulerDriver, offers []*mesos.Offer) {
fmt.Println("BPSW scheduling...")
baseSchedRef := spc.(*baseScheduler)
baseSchedRef := spc.(*BaseScheduler)
def.SortTasks(baseSchedRef.tasks, def.SortByWatts)
baseSchedRef.LogOffersReceived(offers)