custom-kube-scheduler/kubernetes-v1.15.4/pkg/scheduler/customcache/cache.go

package customcache

import (
	"sync"
)

//var timeout *time.Ticker = time.NewTicker(time.Duration(10 * time.Second))

type MlabCache struct {
	Cache map[string]map[string]float64
	Mux   sync.Mutex
}

var LabCache MlabCache = MlabCache{
	Cache: map[string]map[string]float64{
		"kube-01": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-02": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-03": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-04": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-05": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-06": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-07": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-08": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
	},
}

func (c *MlabCache) CleanCache() {
	c.Mux.Lock()
	c.Cache = map[string]map[string]float64{
		"kube-01": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-02": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-03": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-04": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-05": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-06": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-07": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
		"kube-08": map[string]float64{
			"ipc":       -1,
			"mem_read":  -1,
			"mem_write": -1,
			"c6res":     -1,
		},
	}
	c.Mux.Unlock()
}

func (c *MlabCache) UpdateCache(input map[string]float64, c6res float64, nodename string) error {
	c.Mux.Lock()
	c.Cache[nodename] = map[string]float64{
		"ipc":       input["ipc"],
		"mem_read":  input["mem_read"],
		"mem_write": input["mem_write"],
		"c6res":     input["c6res"],
	}
	c.Mux.Unlock()

	return nil
}

func (c *MlabCache) AddAppMetrics(app map[string]float64, nodename string, numCores int) {
	c.Mux.Lock()
	c.Cache[nodename]["mem_read"] += app["mem_read"]
	c.Cache[nodename]["mem_write"] += app["mem_write"]
	//TODO
	// handle c6res addition
	c.Cache[nodename]["c6res"] -= (1 - app["c6res"]) / float64(100*numCores)

	//TODO
	// handle ipc addition
	c.Mux.Unlock()
}