cpu.go

package hostInfo

import (
	"fmt"
	"github.com/pkg/errors"
	"github.com/rancher/agent/utilities/constants"
	"os"
)

type CPUCollector struct {
}

func (c CPUCollector) GetData() (map[string]interface{}, error) {
	data := map[string]interface{}{}

	cInfo, err := c.getCPUInfo()
	if err != nil {
		return data, errors.Wrap(err, constants.CPUGetDataError+"failed to get cpu info")
	}
	percent, err := c.getCPUPercentage()
	if err != nil {
		return data, errors.Wrap(err, constants.CPUGetDataError+"failed to get cpu percentage")
	}
	for key, value := range percent {
		data[key] = value
	}
	for key, value := range cInfo {
		data[key] = value
	}
	for key, value := range c.getCPULoadAverage() {
		data[key] = value
	}
	return data, nil
}

func (c CPUCollector) KeyName() string {
	return "cpuInfo"
}

func (c CPUCollector) GetLabels(prefix string) (map[string]string, error) {
	if _, err := os.Stat("/dev/kvm"); err == nil {
		return map[string]string{
			fmt.Sprintf("%s.%s", prefix, "kvm"): "true",
		}, nil
	}
	return map[string]string{}, nil
}