open-nomad/client/fingerprint/host.go

54 lines
1.5 KiB
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package fingerprint
import (
"runtime"
"strings"
log "github.com/hashicorp/go-hclog"
"github.com/shirou/gopsutil/v3/host"
)
// HostFingerprint is used to fingerprint the host
type HostFingerprint struct {
StaticFingerprinter
logger log.Logger
}
// NewHostFingerprint is used to create a Host fingerprint
func NewHostFingerprint(logger log.Logger) Fingerprint {
f := &HostFingerprint{logger: logger.Named("host")}
return f
}
func (f *HostFingerprint) Fingerprint(req *FingerprintRequest, resp *FingerprintResponse) error {
hostInfo, err := host.Info()
if err != nil {
f.logger.Warn("error retrieving host information", "error", err)
return err
}
if runtime.GOOS == "windows" {
platformVersion := strings.Split(hostInfo.PlatformVersion, "Build")
if len(platformVersion) == 2 {
resp.AddAttribute("os.version", strings.TrimSpace(platformVersion[0]))
resp.AddAttribute("os.build", strings.TrimSpace(platformVersion[1]))
} else {
f.logger.Warn("unable to retrieve 'os.build' attribute", "platform_version", hostInfo.PlatformVersion)
}
} else {
resp.AddAttribute("os.version", hostInfo.PlatformVersion)
resp.AddAttribute("kernel.version", hostInfo.KernelVersion)
}
resp.AddAttribute("os.name", hostInfo.Platform)
resp.AddAttribute("kernel.name", runtime.GOOS)
resp.AddAttribute("kernel.arch", hostInfo.KernelArch)
resp.AddAttribute("unique.hostname", hostInfo.Hostname)
resp.Detected = true
return nil
}