metric_sys.go

package minivmm

import (
	"github.com/mackerelio/go-osstat/cpu"
	"github.com/mackerelio/go-osstat/memory"
	"golang.org/x/sys/unix"
)

// SysMetric is the metrics of system resrouces.
type SysMetric struct {
	CPUCores        int    `json:"minivmm_sys_cpu_cores"`
	MemoryBytes     uint64 `json:"minivmm_sys_memory_bytes"`
	MemoryBytesUsed uint64 `json:"minivmm_sys_memory_bytes_used"`
	DiskBytes       uint64 `json:"minivmm_sys_disk_bytes"`
	DiskBytesUsed   uint64 `json:"minivmm_sys_disk_bytes_used"`
}

// GetSysMetric returns the system resource metrics.
func GetSysMetric() (*SysMetric, error) {
	var m SysMetric

	cpu, err := cpu.Get()
	if err != nil {
		return nil, err
	}
	m.CPUCores = cpu.CPUCount

	memory, err := memory.Get()
	if err != nil {
		return nil, err
	}
	m.MemoryBytes = memory.Total
	m.MemoryBytesUsed = memory.Used

	m.DiskBytes = getDiskSizeTotal(C.Dir)
	m.DiskBytesUsed = getDiskSizeUsed(C.Dir)

	return &m, nil
}

func getDiskSizeTotal(path string) uint64 {
	var stat unix.Statfs_t
	unix.Statfs(path, &stat)

	return stat.Blocks * uint64(stat.Bsize)
}

func getDiskSizeUsed(path string) uint64 {
	var stat unix.Statfs_t
	unix.Statfs(path, &stat)

	return (stat.Blocks - stat.Bavail) * uint64(stat.Bsize)
}