mirror of
https://github.com/kemko/nomad.git
synced 2026-01-07 10:55:42 +03:00
591394fb62
* drivers: plumb hardware topology via grpc into drivers This PR swaps out the temporary use of detecting system hardware manually in each driver for using the Client's detected topology by plumbing the data over gRPC. This ensures that Client configuration is taken to account consistently in all references to system topology. * cr: use enum instead of bool for core grade * cr: fix test slit tables to be possible
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
//go:build !linux && !darwin
|
|
|
|
package numalib
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/client/lib/idset"
|
|
"github.com/hashicorp/nomad/client/lib/numalib/hw"
|
|
"github.com/shirou/gopsutil/v3/cpu"
|
|
)
|
|
|
|
// PlatformScanners returns the set of SystemScanner for systems without a
|
|
// specific implementation.
|
|
func PlatformScanners() []SystemScanner {
|
|
return []SystemScanner{
|
|
new(Generic),
|
|
}
|
|
}
|
|
|
|
const (
|
|
nodeID = hw.NodeID(0)
|
|
socketID = hw.SocketID(0)
|
|
maxSpeed = hw.KHz(0)
|
|
)
|
|
|
|
// Generic implements SystemScanner as a fallback for operating systems without
|
|
// a specific implementation.
|
|
type Generic struct{}
|
|
|
|
func (g *Generic) ScanSystem(top *Topology) {
|
|
// hardware may or may not be NUMA, but for now we only
|
|
// detect such topology on linux systems
|
|
top.NodeIDs = idset.Empty[hw.NodeID]()
|
|
top.NodeIDs.Insert(nodeID)
|
|
|
|
// cores
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
count, err := cpu.CountsWithContext(ctx, true)
|
|
if err != nil {
|
|
return
|
|
}
|
|
top.Cores = make([]Core, count)
|
|
|
|
infos, err := cpu.InfoWithContext(ctx)
|
|
if err != nil || len(infos) == 0 {
|
|
return
|
|
}
|
|
|
|
for i := 0; i < count; i++ {
|
|
info := infos[0]
|
|
speed := hw.KHz(hw.MHz(info.Mhz) * 1000)
|
|
top.insert(nodeID, socketID, hw.CoreID(i), Performance, maxSpeed, speed)
|
|
}
|
|
}
|