smarter-device-manager/argus.go

36 lines
593 B
Go

// Copyright (c) 2019, Arm Ltd
package main
import (
"github.com/golang/glog"
pluginapi "k8s.io/kubelet/pkg/apis/deviceplugin/v1beta1"
)
func check(err error) {
if err != nil {
glog.Errorf(err.Error())
}
}
func getDevices(n uint) []*pluginapi.Device {
var devs []*pluginapi.Device
for i := uint(0); i < n; i++ {
devs = append(devs, &pluginapi.Device{
ID: string(i),
Health: pluginapi.Healthy,
})
}
return devs
}
func deviceExists(devs []*pluginapi.Device, id string) bool {
for _, d := range devs {
if d.ID == id {
return true
}
}
return false
}