Created utilities/offerUtils that can can hold all the utility functions for Offers.
This commit is contained in:
parent
6d4446413d
commit
e63784b007
1 changed files with 40 additions and 0 deletions
40
utilities/offerUtils/offerUtils.go
Normal file
40
utilities/offerUtils/offerUtils.go
Normal file
|
@ -0,0 +1,40 @@
|
|||
package offerUtils
|
||||
|
||||
import (
|
||||
mesos "github.com/mesos/mesos-go/mesosproto"
|
||||
)
|
||||
|
||||
func OfferAgg(offer *mesos.Offer) (float64, float64, float64) {
|
||||
var cpus, mem, watts float64
|
||||
|
||||
for _, resource := range offer.Resources {
|
||||
switch resource.GetName() {
|
||||
case "cpus":
|
||||
cpus += *resource.GetScalar().Value
|
||||
case "mem":
|
||||
mem += *resource.GetScalar().Value
|
||||
case "watts":
|
||||
watts += *resource.GetScalar().Value
|
||||
}
|
||||
}
|
||||
|
||||
return cpus, mem, watts
|
||||
}
|
||||
|
||||
type OffersSorter []*mesos.Offer
|
||||
|
||||
func (offersSorter OffersSorter) Len() int {
|
||||
return len(offersSorter)
|
||||
}
|
||||
|
||||
func (offersSorter OffersSorter) Swap(i, j int) {
|
||||
offersSorter[i], offersSorter[j] = offersSorter[j], offersSorter[i]
|
||||
}
|
||||
|
||||
func (offersSorter OffersSorter) Less(i, j int) bool {
|
||||
// getting CPU resource availability of offersSorter[i]
|
||||
cpu1, _, _ := OfferAgg(offersSorter[i])
|
||||
// getting CPU resource availability of offersSorter[j]
|
||||
cpu2, _, _ := OfferAgg(offersSorter[j])
|
||||
return cpu1 <= cpu2
|
||||
}
|
Reference in a new issue