mirror of
https://github.com/kemko/nomad.git
synced 2026-01-06 18:35:44 +03:00
b8397a712d
on Linux systems this is derived from the configure cpuset cgroup parent (defaults to /nomad) for non Linux systems and Linux systems where cgroups are not enabled, the client defaults to using all cores
106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package fingerprint
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/nomad/lib/cpuset"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/nomad/helper/stats"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
)
|
|
|
|
const (
|
|
// defaultCPUTicks is the default amount of CPU resources assumed to be
|
|
// available if the CPU performance data is unable to be detected. This is
|
|
// common on EC2 instances, where the env_aws fingerprinter will follow up,
|
|
// setting an accurate value.
|
|
defaultCPUTicks = 1000 // 1 core * 1 GHz
|
|
)
|
|
|
|
// CPUFingerprint is used to fingerprint the CPU
|
|
type CPUFingerprint struct {
|
|
StaticFingerprinter
|
|
logger log.Logger
|
|
}
|
|
|
|
// NewCPUFingerprint is used to create a CPU fingerprint
|
|
func NewCPUFingerprint(logger log.Logger) Fingerprint {
|
|
f := &CPUFingerprint{logger: logger.Named("cpu")}
|
|
return f
|
|
}
|
|
|
|
func (f *CPUFingerprint) Fingerprint(req *FingerprintRequest, resp *FingerprintResponse) error {
|
|
cfg := req.Config
|
|
setResourcesCPU := func(totalCompute int, totalCores uint16, reservableCores []uint16) {
|
|
// COMPAT(0.10): Remove in 0.10
|
|
resp.Resources = &structs.Resources{
|
|
CPU: totalCompute,
|
|
}
|
|
|
|
resp.NodeResources = &structs.NodeResources{
|
|
Cpu: structs.NodeCpuResources{
|
|
CpuShares: int64(totalCompute),
|
|
TotalCpuCores: totalCores,
|
|
ReservableCpuCores: reservableCores,
|
|
},
|
|
}
|
|
}
|
|
|
|
if err := stats.Init(); err != nil {
|
|
f.logger.Warn("failed initializing stats collector", "error", err)
|
|
}
|
|
|
|
if modelName := stats.CPUModelName(); modelName != "" {
|
|
resp.AddAttribute("cpu.modelname", modelName)
|
|
}
|
|
|
|
if mhz := stats.CPUMHzPerCore(); mhz > 0 {
|
|
resp.AddAttribute("cpu.frequency", fmt.Sprintf("%.0f", mhz))
|
|
f.logger.Debug("detected cpu frequency", "MHz", log.Fmt("%.0f", mhz))
|
|
}
|
|
|
|
var numCores int
|
|
if numCores = stats.CPUNumCores(); numCores > 0 {
|
|
resp.AddAttribute("cpu.numcores", fmt.Sprintf("%d", numCores))
|
|
f.logger.Debug("detected core count", "cores", numCores)
|
|
}
|
|
|
|
var reservableCores []uint16
|
|
if cores, err := f.deriveReservableCores(req, numCores); err != nil {
|
|
f.logger.Warn("failed to detect set of reservable cores", "error", err)
|
|
} else {
|
|
reservableCores = cpuset.New(cores...).Difference(cpuset.New(req.Node.ReservedResources.Cpu.ReservedCpuCores...)).ToSlice()
|
|
f.logger.Debug("detected reservable cores", "cpuset", reservableCores)
|
|
}
|
|
|
|
tt := int(stats.TotalTicksAvailable())
|
|
if cfg.CpuCompute > 0 {
|
|
f.logger.Debug("using user specified cpu compute", "cpu_compute", cfg.CpuCompute)
|
|
tt = cfg.CpuCompute
|
|
}
|
|
|
|
// If we cannot detect the cpu total compute, fallback to a very low default
|
|
// value and log a message about configuring cpu_total_compute. This happens
|
|
// on Graviton instances where CPU information is unavailable. In that case,
|
|
// the env_aws fingerprinter updates the value with correct information.
|
|
if tt == 0 {
|
|
f.logger.Info("fallback to default cpu total compute, set client config option cpu_total_compute to override")
|
|
tt = defaultCPUTicks
|
|
}
|
|
|
|
resp.AddAttribute("cpu.totalcompute", fmt.Sprintf("%d", tt))
|
|
setResourcesCPU(tt, uint16(numCores), reservableCores)
|
|
resp.Detected = true
|
|
|
|
return nil
|
|
}
|
|
|
|
func defaultReservableCores(totalCores int) []uint16 {
|
|
cores := make([]uint16, totalCores)
|
|
for i := range cores {
|
|
cores[i] = uint16(i)
|
|
}
|
|
return cores
|
|
}
|