50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package fingerprint
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os/exec"
|
|
"runtime"
|
|
|
|
"github.com/hashicorp/nomad/client/config"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
"github.com/shirou/gopsutil/host"
|
|
)
|
|
|
|
// HostFingerprint is used to fingerprint the host
|
|
type HostFingerprint struct {
|
|
logger *log.Logger
|
|
}
|
|
|
|
// NewHostFingerprint is used to create a Host fingerprint
|
|
func NewHostFingerprint(logger *log.Logger) Fingerprint {
|
|
f := &HostFingerprint{logger: logger}
|
|
return f
|
|
}
|
|
|
|
func (f *HostFingerprint) Fingerprint(cfg *config.Config, node *structs.Node) (bool, error) {
|
|
hostInfo, err := host.HostInfo()
|
|
if err != nil {
|
|
f.logger.Println("[WARN] Error retrieving host information: ", err)
|
|
return false, err
|
|
}
|
|
|
|
node.Attributes["os.name"] = hostInfo.Platform
|
|
node.Attributes["os.version"] = hostInfo.PlatformVersion
|
|
|
|
node.Attributes["kernel.name"] = runtime.GOOS
|
|
node.Attributes["kernel.version"] = ""
|
|
|
|
if runtime.GOOS != "windows" {
|
|
out, err := exec.Command("uname", "-r").Output()
|
|
if err != nil {
|
|
return false, fmt.Errorf("Failed to run uname: %s", err)
|
|
}
|
|
node.Attributes["kernel.version"] = string(out)
|
|
}
|
|
|
|
node.Attributes["hostname"] = hostInfo.Hostname
|
|
|
|
return true, nil
|
|
}
|