102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
|
package hostutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/hashicorp/go-multierror"
|
||
|
|
"github.com/shirou/gopsutil/cpu"
|
||
|
|
"github.com/shirou/gopsutil/disk"
|
||
|
|
"github.com/shirou/gopsutil/host"
|
||
|
|
"github.com/shirou/gopsutil/mem"
|
||
|
|
)
|
||
|
|
|
||
|
|
// HostInfo holds all the information that gets captured on the host. The
|
||
|
|
// set of information captured depends on the host operating system. For more
|
||
|
|
// information, refer to: https://github.com/shirou/gopsutil#current-status
|
||
|
|
type HostInfo struct {
|
||
|
|
// Timestamp returns the timestamp in UTC on the collection time.
|
||
|
|
Timestamp time.Time `json:"timestamp"`
|
||
|
|
// CPU returns information about the CPU such as family, model, cores, etc.
|
||
|
|
CPU []cpu.InfoStat `json:"cpu"`
|
||
|
|
// CPUTimes returns statistics on CPU usage represented in Jiffies.
|
||
|
|
CPUTimes []cpu.TimesStat `json:"cpu_times"`
|
||
|
|
// Disk returns statitics on disk usage for all accessible partitions.
|
||
|
|
Disk []*disk.UsageStat `json:"disk"`
|
||
|
|
// Host returns general host information such as hostname, platform, uptime,
|
||
|
|
// kernel version, etc.
|
||
|
|
Host *host.InfoStat `json:"host"`
|
||
|
|
// Memory contains statistics about the memory such as total, available, and
|
||
|
|
// used memory in number of bytes.
|
||
|
|
Memory *mem.VirtualMemoryStat `json:"memory"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// HostInfoError is a typed error for more convenient error checking.
|
||
|
|
type HostInfoError struct {
|
||
|
|
Type string
|
||
|
|
Err error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *HostInfoError) WrappedErrors() []error {
|
||
|
|
return []error{e.Err}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *HostInfoError) Error() string {
|
||
|
|
return fmt.Sprintf("%s: %s", e.Type, e.Err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// CollectHostInfo returns information on the host, which includes general
|
||
|
|
// host status, CPU, memory, and disk utilization.
|
||
|
|
//
|
||
|
|
// The function does a best-effort capture on the most information possible,
|
||
|
|
// continuing on capture errors encountered and appending them to a resulting
|
||
|
|
// multierror.Error that gets returned at the end.
|
||
|
|
func CollectHostInfo() (*HostInfo, error) {
|
||
|
|
var retErr *multierror.Error
|
||
|
|
info := &HostInfo{Timestamp: time.Now().UTC()}
|
||
|
|
|
||
|
|
if h, err := host.Info(); err != nil {
|
||
|
|
retErr = multierror.Append(retErr, &HostInfoError{"host", err})
|
||
|
|
} else {
|
||
|
|
info.Host = h
|
||
|
|
}
|
||
|
|
|
||
|
|
if v, err := mem.VirtualMemory(); err != nil {
|
||
|
|
retErr = multierror.Append(retErr, &HostInfoError{"memory", err})
|
||
|
|
} else {
|
||
|
|
info.Memory = v
|
||
|
|
}
|
||
|
|
|
||
|
|
parts, err := disk.Partitions(false)
|
||
|
|
if err != nil {
|
||
|
|
retErr = multierror.Append(retErr, &HostInfoError{"disk", err})
|
||
|
|
} else {
|
||
|
|
var usage []*disk.UsageStat
|
||
|
|
for i, part := range parts {
|
||
|
|
u, err := disk.Usage(part.Mountpoint)
|
||
|
|
if err != nil {
|
||
|
|
retErr = multierror.Append(retErr, &HostInfoError{fmt.Sprintf("disk.%d", i), err})
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
usage = append(usage, u)
|
||
|
|
|
||
|
|
}
|
||
|
|
info.Disk = usage
|
||
|
|
}
|
||
|
|
|
||
|
|
if c, err := cpu.Info(); err != nil {
|
||
|
|
retErr = multierror.Append(retErr, &HostInfoError{"cpu", err})
|
||
|
|
} else {
|
||
|
|
info.CPU = c
|
||
|
|
}
|
||
|
|
|
||
|
|
t, err := cpu.Times(true)
|
||
|
|
if err != nil {
|
||
|
|
retErr = multierror.Append(retErr, &HostInfoError{"cpu_times", err})
|
||
|
|
} else {
|
||
|
|
info.CPUTimes = t
|
||
|
|
}
|
||
|
|
|
||
|
|
return info, retErr.ErrorOrNil()
|
||
|
|
}
|