2015-08-26 21:27:44 +00:00
|
|
|
package fingerprint
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
2015-08-28 00:37:37 +00:00
|
|
|
"runtime"
|
2015-08-26 21:27:44 +00:00
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
cstructs "github.com/hashicorp/nomad/client/structs"
|
2015-08-26 21:27:44 +00:00
|
|
|
"github.com/shirou/gopsutil/host"
|
|
|
|
)
|
|
|
|
|
|
|
|
// HostFingerprint is used to fingerprint the host
|
|
|
|
type HostFingerprint struct {
|
2015-11-05 21:46:02 +00:00
|
|
|
StaticFingerprinter
|
2015-08-26 21:27:44 +00:00
|
|
|
logger *log.Logger
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewHostFingerprint is used to create a Host fingerprint
|
|
|
|
func NewHostFingerprint(logger *log.Logger) Fingerprint {
|
2015-08-27 00:16:34 +00:00
|
|
|
f := &HostFingerprint{logger: logger}
|
2015-08-26 21:27:44 +00:00
|
|
|
return f
|
|
|
|
}
|
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
func (f *HostFingerprint) Fingerprint(req *cstructs.FingerprintRequest, resp *cstructs.FingerprintResponse) error {
|
2016-05-09 15:19:04 +00:00
|
|
|
hostInfo, err := host.Info()
|
2015-08-26 21:27:44 +00:00
|
|
|
if err != nil {
|
|
|
|
f.logger.Println("[WARN] Error retrieving host information: ", err)
|
2018-01-24 14:09:53 +00:00
|
|
|
return err
|
2015-08-26 21:27:44 +00:00
|
|
|
}
|
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
resp.Attributes["os.name"] = hostInfo.Platform
|
|
|
|
resp.Attributes["os.version"] = hostInfo.PlatformVersion
|
2015-08-28 00:37:37 +00:00
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
resp.Attributes["kernel.name"] = runtime.GOOS
|
|
|
|
resp.Attributes["kernel.version"] = hostInfo.KernelVersion
|
2015-08-28 00:37:37 +00:00
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
resp.Attributes["unique.hostname"] = hostInfo.Hostname
|
2015-08-26 21:27:44 +00:00
|
|
|
|
2018-01-24 14:09:53 +00:00
|
|
|
return nil
|
2015-08-26 21:27:44 +00:00
|
|
|
}
|