Introduce canonical hardware.devices repository and align UI/Reanimator exports
This commit is contained in:
734
internal/server/device_repository.go
Normal file
734
internal/server/device_repository.go
Normal file
@@ -0,0 +1,734 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"git.mchus.pro/mchus/logpile/internal/models"
|
||||
)
|
||||
|
||||
type slotFirmwareInfo struct {
|
||||
Model string
|
||||
Version string
|
||||
Category string
|
||||
}
|
||||
|
||||
var (
|
||||
psuFirmwareRe = regexp.MustCompile(`(?i)^PSU\s*([0-9A-Za-z_-]+)\s*(?:\(([^)]+)\))?$`)
|
||||
nicFirmwareRe = regexp.MustCompile(`(?i)^NIC\s+([^()]+?)\s*(?:\(([^)]+)\))?$`)
|
||||
gpuFirmwareRe = regexp.MustCompile(`(?i)^GPU\s+([^()]+?)\s*(?:\(([^)]+)\))?$`)
|
||||
nvsFirmwareRe = regexp.MustCompile(`(?i)^NVSwitch\s+([^()]+?)\s*(?:\(([^)]+)\))?$`)
|
||||
)
|
||||
|
||||
func BuildHardwareDevices(hw *models.HardwareConfig) []models.HardwareDevice {
|
||||
if hw == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
all := make([]models.HardwareDevice, 0, 1+len(hw.CPUs)+len(hw.Memory)+len(hw.Storage)+len(hw.PCIeDevices)+len(hw.GPUs)+len(hw.NetworkAdapters)+len(hw.PowerSupply))
|
||||
fwBySlot := buildFirmwareBySlot(hw.Firmware)
|
||||
nextID := 0
|
||||
add := func(d models.HardwareDevice) {
|
||||
d.ID = fmt.Sprintf("%s:%d", d.Kind, nextID)
|
||||
nextID++
|
||||
all = append(all, d)
|
||||
}
|
||||
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindBoard,
|
||||
Source: "board",
|
||||
Slot: "board",
|
||||
Model: strings.TrimSpace(hw.BoardInfo.ProductName),
|
||||
PartNumber: strings.TrimSpace(hw.BoardInfo.PartNumber),
|
||||
Manufacturer: strings.TrimSpace(hw.BoardInfo.Manufacturer),
|
||||
SerialNumber: strings.TrimSpace(hw.BoardInfo.SerialNumber),
|
||||
Details: map[string]any{
|
||||
"description": strings.TrimSpace(hw.BoardInfo.Description),
|
||||
"version": strings.TrimSpace(hw.BoardInfo.Version),
|
||||
"uuid": strings.TrimSpace(hw.BoardInfo.UUID),
|
||||
},
|
||||
})
|
||||
|
||||
for _, cpu := range hw.CPUs {
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindCPU,
|
||||
Source: "cpus",
|
||||
Slot: fmt.Sprintf("CPU%d", cpu.Socket),
|
||||
Model: cpu.Model,
|
||||
SerialNumber: cpu.SerialNumber,
|
||||
Cores: cpu.Cores,
|
||||
Threads: cpu.Threads,
|
||||
FrequencyMHz: cpu.FrequencyMHz,
|
||||
MaxFreqMHz: cpu.MaxFreqMHz,
|
||||
Status: cpu.Status,
|
||||
StatusCheckedAt: cpu.StatusCheckedAt,
|
||||
StatusChangedAt: cpu.StatusChangedAt,
|
||||
StatusAtCollect: cpu.StatusAtCollect,
|
||||
StatusHistory: cpu.StatusHistory,
|
||||
ErrorDescription: cpu.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": cpu.Description,
|
||||
"socket": cpu.Socket,
|
||||
"l1_cache_kb": cpu.L1CacheKB,
|
||||
"l2_cache_kb": cpu.L2CacheKB,
|
||||
"l3_cache_kb": cpu.L3CacheKB,
|
||||
"tdp_w": cpu.TDP,
|
||||
"ppin": cpu.PPIN,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
for _, mem := range hw.Memory {
|
||||
if !mem.Present || mem.SizeMB == 0 {
|
||||
continue
|
||||
}
|
||||
present := mem.Present
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindMemory,
|
||||
Source: "memory",
|
||||
Slot: mem.Slot,
|
||||
Location: mem.Location,
|
||||
Manufacturer: mem.Manufacturer,
|
||||
SerialNumber: mem.SerialNumber,
|
||||
PartNumber: mem.PartNumber,
|
||||
Type: mem.Type,
|
||||
Present: &present,
|
||||
SizeMB: mem.SizeMB,
|
||||
Status: mem.Status,
|
||||
StatusCheckedAt: mem.StatusCheckedAt,
|
||||
StatusChangedAt: mem.StatusChangedAt,
|
||||
StatusAtCollect: mem.StatusAtCollect,
|
||||
StatusHistory: mem.StatusHistory,
|
||||
ErrorDescription: mem.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": mem.Description,
|
||||
"technology": mem.Technology,
|
||||
"max_speed_mhz": mem.MaxSpeedMHz,
|
||||
"current_speed_mhz": mem.CurrentSpeedMHz,
|
||||
"ranks": mem.Ranks,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
for _, stor := range hw.Storage {
|
||||
if !stor.Present {
|
||||
continue
|
||||
}
|
||||
present := stor.Present
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindStorage,
|
||||
Source: "storage",
|
||||
Slot: stor.Slot,
|
||||
Location: stor.Location,
|
||||
Model: stor.Model,
|
||||
Manufacturer: stor.Manufacturer,
|
||||
SerialNumber: stor.SerialNumber,
|
||||
Firmware: stor.Firmware,
|
||||
Type: stor.Type,
|
||||
Interface: stor.Interface,
|
||||
Present: &present,
|
||||
SizeGB: stor.SizeGB,
|
||||
Status: stor.Status,
|
||||
StatusCheckedAt: stor.StatusCheckedAt,
|
||||
StatusChangedAt: stor.StatusChangedAt,
|
||||
StatusAtCollect: stor.StatusAtCollect,
|
||||
StatusHistory: stor.StatusHistory,
|
||||
ErrorDescription: stor.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": stor.Description,
|
||||
"backplane_id": stor.BackplaneID,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
for _, p := range hw.PCIeDevices {
|
||||
if isEmptyPCIeDevice(p) {
|
||||
continue
|
||||
}
|
||||
slotKey := normalizeSlotKey(p.Slot)
|
||||
fwInfo := fwBySlot[slotKey]
|
||||
model := strings.TrimSpace(p.PartNumber)
|
||||
if model == "" {
|
||||
model = strings.TrimSpace(p.DeviceClass)
|
||||
}
|
||||
if model == "" {
|
||||
model = strings.TrimSpace(p.Description)
|
||||
}
|
||||
if model == "" && fwInfo.Model != "" {
|
||||
model = fwInfo.Model
|
||||
}
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindPCIe,
|
||||
Source: "pcie_devices",
|
||||
Slot: p.Slot,
|
||||
BDF: p.BDF,
|
||||
DeviceClass: p.DeviceClass,
|
||||
VendorID: p.VendorID,
|
||||
DeviceID: p.DeviceID,
|
||||
Model: model,
|
||||
PartNumber: p.PartNumber,
|
||||
Manufacturer: p.Manufacturer,
|
||||
SerialNumber: p.SerialNumber,
|
||||
Firmware: fwInfo.Version,
|
||||
MACAddresses: p.MACAddresses,
|
||||
LinkWidth: p.LinkWidth,
|
||||
LinkSpeed: p.LinkSpeed,
|
||||
MaxLinkWidth: p.MaxLinkWidth,
|
||||
MaxLinkSpeed: p.MaxLinkSpeed,
|
||||
Status: p.Status,
|
||||
StatusCheckedAt: p.StatusCheckedAt,
|
||||
StatusChangedAt: p.StatusChangedAt,
|
||||
StatusAtCollect: p.StatusAtCollect,
|
||||
StatusHistory: p.StatusHistory,
|
||||
ErrorDescription: p.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": p.Description,
|
||||
"fw_category": fwInfo.Category,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
for _, gpu := range hw.GPUs {
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindGPU,
|
||||
Source: "gpus",
|
||||
Slot: gpu.Slot,
|
||||
Location: gpu.Location,
|
||||
BDF: gpu.BDF,
|
||||
DeviceClass: "DisplayController",
|
||||
VendorID: gpu.VendorID,
|
||||
DeviceID: gpu.DeviceID,
|
||||
Model: gpu.Model,
|
||||
PartNumber: gpu.PartNumber,
|
||||
Manufacturer: gpu.Manufacturer,
|
||||
SerialNumber: gpu.SerialNumber,
|
||||
Firmware: gpu.Firmware,
|
||||
LinkWidth: gpu.CurrentLinkWidth,
|
||||
LinkSpeed: gpu.CurrentLinkSpeed,
|
||||
MaxLinkWidth: gpu.MaxLinkWidth,
|
||||
MaxLinkSpeed: gpu.MaxLinkSpeed,
|
||||
Status: gpu.Status,
|
||||
StatusCheckedAt: gpu.StatusCheckedAt,
|
||||
StatusChangedAt: gpu.StatusChangedAt,
|
||||
StatusAtCollect: gpu.StatusAtCollect,
|
||||
StatusHistory: gpu.StatusHistory,
|
||||
ErrorDescription: gpu.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": gpu.Description,
|
||||
"uuid": gpu.UUID,
|
||||
"video_bios": gpu.VideoBIOS,
|
||||
"irq": gpu.IRQ,
|
||||
"bus_type": gpu.BusType,
|
||||
"dma_size": gpu.DMASize,
|
||||
"dma_mask": gpu.DMAMask,
|
||||
"device_minor": gpu.DeviceMinor,
|
||||
"temperature": gpu.Temperature,
|
||||
"mem_temperature": gpu.MemTemperature,
|
||||
"power": gpu.Power,
|
||||
"max_power": gpu.MaxPower,
|
||||
"clock_speed": gpu.ClockSpeed,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
for _, nic := range hw.NetworkAdapters {
|
||||
if !nic.Present {
|
||||
continue
|
||||
}
|
||||
present := nic.Present
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindNetwork,
|
||||
Source: "network_adapters",
|
||||
Slot: nic.Slot,
|
||||
Location: nic.Location,
|
||||
VendorID: nic.VendorID,
|
||||
DeviceID: nic.DeviceID,
|
||||
Model: nic.Model,
|
||||
PartNumber: nic.PartNumber,
|
||||
Manufacturer: nic.Vendor,
|
||||
SerialNumber: nic.SerialNumber,
|
||||
Firmware: nic.Firmware,
|
||||
PortCount: nic.PortCount,
|
||||
PortType: nic.PortType,
|
||||
MACAddresses: nic.MACAddresses,
|
||||
Present: &present,
|
||||
Status: nic.Status,
|
||||
StatusCheckedAt: nic.StatusCheckedAt,
|
||||
StatusChangedAt: nic.StatusChangedAt,
|
||||
StatusAtCollect: nic.StatusAtCollect,
|
||||
StatusHistory: nic.StatusHistory,
|
||||
ErrorDescription: nic.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": nic.Description,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
for _, psu := range hw.PowerSupply {
|
||||
if !psu.Present {
|
||||
continue
|
||||
}
|
||||
present := psu.Present
|
||||
add(models.HardwareDevice{
|
||||
Kind: models.DeviceKindPSU,
|
||||
Source: "power_supplies",
|
||||
Slot: psu.Slot,
|
||||
Model: psu.Model,
|
||||
PartNumber: psu.PartNumber,
|
||||
Manufacturer: psu.Vendor,
|
||||
SerialNumber: psu.SerialNumber,
|
||||
Firmware: psu.Firmware,
|
||||
Present: &present,
|
||||
WattageW: psu.WattageW,
|
||||
InputType: psu.InputType,
|
||||
InputPowerW: psu.InputPowerW,
|
||||
OutputPowerW: psu.OutputPowerW,
|
||||
InputVoltage: psu.InputVoltage,
|
||||
TemperatureC: psu.TemperatureC,
|
||||
Status: psu.Status,
|
||||
StatusCheckedAt: psu.StatusCheckedAt,
|
||||
StatusChangedAt: psu.StatusChangedAt,
|
||||
StatusAtCollect: psu.StatusAtCollect,
|
||||
StatusHistory: psu.StatusHistory,
|
||||
ErrorDescription: psu.ErrorDescription,
|
||||
Details: map[string]any{
|
||||
"description": psu.Description,
|
||||
"output_voltage": psu.OutputVoltage,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
return dedupeDevices(all)
|
||||
}
|
||||
|
||||
func isEmptyPCIeDevice(p models.PCIeDevice) bool {
|
||||
if isNumericSlot(strings.TrimSpace(p.Slot)) &&
|
||||
strings.TrimSpace(p.BDF) == "" &&
|
||||
p.VendorID == 0 &&
|
||||
p.DeviceID == 0 &&
|
||||
normalizedSerial(p.SerialNumber) == "" &&
|
||||
!hasMeaningfulText(p.PartNumber) &&
|
||||
!hasMeaningfulText(p.Manufacturer) &&
|
||||
!hasMeaningfulText(p.Description) &&
|
||||
len(p.MACAddresses) == 0 &&
|
||||
p.LinkWidth == 0 &&
|
||||
p.MaxLinkWidth == 0 {
|
||||
return true
|
||||
}
|
||||
|
||||
if strings.TrimSpace(p.BDF) != "" {
|
||||
return false
|
||||
}
|
||||
if p.VendorID != 0 || p.DeviceID != 0 {
|
||||
return false
|
||||
}
|
||||
if normalizedSerial(p.SerialNumber) != "" {
|
||||
return false
|
||||
}
|
||||
if hasMeaningfulText(p.PartNumber) {
|
||||
return false
|
||||
}
|
||||
if hasMeaningfulText(p.Manufacturer) {
|
||||
return false
|
||||
}
|
||||
if hasMeaningfulText(p.Description) {
|
||||
return false
|
||||
}
|
||||
if strings.TrimSpace(p.DeviceClass) != "" {
|
||||
class := strings.ToLower(strings.TrimSpace(p.DeviceClass))
|
||||
if class != "unknown" && class != "other" && class != "pcie device" {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func isNumericSlot(slot string) bool {
|
||||
if slot == "" {
|
||||
return false
|
||||
}
|
||||
for _, r := range slot {
|
||||
if r < '0' || r > '9' {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func hasMeaningfulText(v string) bool {
|
||||
s := strings.ToLower(strings.TrimSpace(v))
|
||||
if s == "" {
|
||||
return false
|
||||
}
|
||||
switch s {
|
||||
case "-", "n/a", "na", "none", "null", "unknown":
|
||||
return false
|
||||
default:
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
func dedupeDevices(items []models.HardwareDevice) []models.HardwareDevice {
|
||||
if len(items) < 2 {
|
||||
return items
|
||||
}
|
||||
parent := make([]int, len(items))
|
||||
for i := range parent {
|
||||
parent[i] = i
|
||||
}
|
||||
find := func(x int) int {
|
||||
for parent[x] != x {
|
||||
parent[x] = parent[parent[x]]
|
||||
x = parent[x]
|
||||
}
|
||||
return x
|
||||
}
|
||||
union := func(a, b int) {
|
||||
ra := find(a)
|
||||
rb := find(b)
|
||||
if ra != rb {
|
||||
parent[rb] = ra
|
||||
}
|
||||
}
|
||||
|
||||
for i := 0; i < len(items); i++ {
|
||||
for j := i + 1; j < len(items); j++ {
|
||||
if shouldMergeDevices(items[i], items[j]) {
|
||||
union(i, j)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
groups := make(map[int][]int, len(items))
|
||||
order := make([]int, 0, len(items))
|
||||
for i := range items {
|
||||
root := find(i)
|
||||
if _, ok := groups[root]; !ok {
|
||||
order = append(order, root)
|
||||
}
|
||||
groups[root] = append(groups[root], i)
|
||||
}
|
||||
|
||||
out := make([]models.HardwareDevice, 0, len(order))
|
||||
for _, root := range order {
|
||||
indices := groups[root]
|
||||
bestIdx := indices[0]
|
||||
bestScore := qualityScore(items[bestIdx])
|
||||
for _, idx := range indices[1:] {
|
||||
if s := qualityScore(items[idx]); s > bestScore {
|
||||
bestIdx = idx
|
||||
bestScore = s
|
||||
}
|
||||
}
|
||||
merged := items[bestIdx]
|
||||
for _, idx := range indices {
|
||||
if idx == bestIdx {
|
||||
continue
|
||||
}
|
||||
merged = mergeDevices(merged, items[idx])
|
||||
}
|
||||
out = append(out, merged)
|
||||
}
|
||||
|
||||
for i := range out {
|
||||
out[i].ID = out[i].Kind + ":" + strconv.Itoa(i)
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func shouldMergeDevices(a, b models.HardwareDevice) bool {
|
||||
aSN := strings.ToLower(normalizedSerial(a.SerialNumber))
|
||||
bSN := strings.ToLower(normalizedSerial(b.SerialNumber))
|
||||
aBDF := strings.ToLower(strings.TrimSpace(a.BDF))
|
||||
bBDF := strings.ToLower(strings.TrimSpace(b.BDF))
|
||||
|
||||
// Hard conflicts.
|
||||
if aSN != "" && bSN != "" && aSN == bSN {
|
||||
return true
|
||||
}
|
||||
if aSN != "" && bSN != "" && aSN != bSN {
|
||||
return false
|
||||
}
|
||||
if aBDF != "" && bBDF != "" && aBDF != bBDF {
|
||||
return false
|
||||
}
|
||||
|
||||
// Strong identities.
|
||||
if aBDF != "" && aBDF == bBDF {
|
||||
return true
|
||||
}
|
||||
|
||||
// If both have no strong IDs, be conservative.
|
||||
if aSN == "" && bSN == "" && aBDF == "" && bBDF == "" {
|
||||
if hasMACOverlap(a.MACAddresses, b.MACAddresses) {
|
||||
return true
|
||||
}
|
||||
if normalizeSlot(a.Slot) != "" && normalizeSlot(a.Slot) == normalizeSlot(b.Slot) {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
score := 0
|
||||
if samePCIID(a, b) {
|
||||
score += 4
|
||||
}
|
||||
if sameModel(a, b) {
|
||||
score += 3
|
||||
}
|
||||
if sameManufacturer(a, b) {
|
||||
score += 2
|
||||
}
|
||||
if normalizeSlot(a.Slot) != "" && normalizeSlot(a.Slot) == normalizeSlot(b.Slot) {
|
||||
score += 2
|
||||
}
|
||||
if hasMACOverlap(a.MACAddresses, b.MACAddresses) {
|
||||
score += 2
|
||||
}
|
||||
if sameKindFamily(a.Kind, b.Kind) {
|
||||
score++
|
||||
}
|
||||
if samePCIID(a, b) && ((aBDF != "" && bBDF == "") || (aBDF == "" && bBDF != "")) {
|
||||
score += 2
|
||||
}
|
||||
|
||||
return score >= 7
|
||||
}
|
||||
|
||||
func mergeDevices(primary, secondary models.HardwareDevice) models.HardwareDevice {
|
||||
fillString := func(dst *string, src string) {
|
||||
if strings.TrimSpace(*dst) == "" && strings.TrimSpace(src) != "" {
|
||||
*dst = src
|
||||
}
|
||||
}
|
||||
fillInt := func(dst *int, src int) {
|
||||
if *dst == 0 && src != 0 {
|
||||
*dst = src
|
||||
}
|
||||
}
|
||||
fillFloat := func(dst *float64, src float64) {
|
||||
if *dst == 0 && src != 0 {
|
||||
*dst = src
|
||||
}
|
||||
}
|
||||
|
||||
fillString(&primary.ID, secondary.ID)
|
||||
fillString(&primary.Kind, secondary.Kind)
|
||||
fillString(&primary.Source, secondary.Source)
|
||||
fillString(&primary.Slot, secondary.Slot)
|
||||
fillString(&primary.Location, secondary.Location)
|
||||
fillString(&primary.BDF, secondary.BDF)
|
||||
fillString(&primary.DeviceClass, secondary.DeviceClass)
|
||||
fillInt(&primary.VendorID, secondary.VendorID)
|
||||
fillInt(&primary.DeviceID, secondary.DeviceID)
|
||||
fillString(&primary.Model, secondary.Model)
|
||||
fillString(&primary.PartNumber, secondary.PartNumber)
|
||||
fillString(&primary.Manufacturer, secondary.Manufacturer)
|
||||
fillString(&primary.SerialNumber, secondary.SerialNumber)
|
||||
fillString(&primary.Firmware, secondary.Firmware)
|
||||
fillString(&primary.Type, secondary.Type)
|
||||
fillString(&primary.Interface, secondary.Interface)
|
||||
if primary.Present == nil && secondary.Present != nil {
|
||||
primary.Present = secondary.Present
|
||||
}
|
||||
fillInt(&primary.SizeMB, secondary.SizeMB)
|
||||
fillInt(&primary.SizeGB, secondary.SizeGB)
|
||||
fillInt(&primary.Cores, secondary.Cores)
|
||||
fillInt(&primary.Threads, secondary.Threads)
|
||||
fillInt(&primary.FrequencyMHz, secondary.FrequencyMHz)
|
||||
fillInt(&primary.MaxFreqMHz, secondary.MaxFreqMHz)
|
||||
fillInt(&primary.PortCount, secondary.PortCount)
|
||||
fillString(&primary.PortType, secondary.PortType)
|
||||
if len(primary.MACAddresses) == 0 && len(secondary.MACAddresses) > 0 {
|
||||
primary.MACAddresses = secondary.MACAddresses
|
||||
}
|
||||
fillInt(&primary.LinkWidth, secondary.LinkWidth)
|
||||
fillString(&primary.LinkSpeed, secondary.LinkSpeed)
|
||||
fillInt(&primary.MaxLinkWidth, secondary.MaxLinkWidth)
|
||||
fillString(&primary.MaxLinkSpeed, secondary.MaxLinkSpeed)
|
||||
fillInt(&primary.WattageW, secondary.WattageW)
|
||||
fillString(&primary.InputType, secondary.InputType)
|
||||
fillInt(&primary.InputPowerW, secondary.InputPowerW)
|
||||
fillInt(&primary.OutputPowerW, secondary.OutputPowerW)
|
||||
fillFloat(&primary.InputVoltage, secondary.InputVoltage)
|
||||
fillInt(&primary.TemperatureC, secondary.TemperatureC)
|
||||
fillString(&primary.Status, secondary.Status)
|
||||
if primary.StatusCheckedAt.IsZero() && !secondary.StatusCheckedAt.IsZero() {
|
||||
primary.StatusCheckedAt = secondary.StatusCheckedAt
|
||||
}
|
||||
if primary.StatusChangedAt.IsZero() && !secondary.StatusChangedAt.IsZero() {
|
||||
primary.StatusChangedAt = secondary.StatusChangedAt
|
||||
}
|
||||
if primary.StatusAtCollect == nil && secondary.StatusAtCollect != nil {
|
||||
primary.StatusAtCollect = secondary.StatusAtCollect
|
||||
}
|
||||
if len(primary.StatusHistory) == 0 && len(secondary.StatusHistory) > 0 {
|
||||
primary.StatusHistory = secondary.StatusHistory
|
||||
}
|
||||
fillString(&primary.ErrorDescription, secondary.ErrorDescription)
|
||||
if primary.Details == nil && secondary.Details != nil {
|
||||
primary.Details = secondary.Details
|
||||
}
|
||||
return primary
|
||||
}
|
||||
|
||||
func samePCIID(a, b models.HardwareDevice) bool {
|
||||
if (a.VendorID == 0 && a.DeviceID == 0) || (b.VendorID == 0 && b.DeviceID == 0) {
|
||||
return false
|
||||
}
|
||||
return a.VendorID == b.VendorID && a.DeviceID == b.DeviceID
|
||||
}
|
||||
|
||||
func sameModel(a, b models.HardwareDevice) bool {
|
||||
am := normalizeText(coalesce(a.Model, a.PartNumber, a.DeviceClass))
|
||||
bm := normalizeText(coalesce(b.Model, b.PartNumber, b.DeviceClass))
|
||||
return am != "" && am == bm
|
||||
}
|
||||
|
||||
func sameManufacturer(a, b models.HardwareDevice) bool {
|
||||
am := normalizeText(a.Manufacturer)
|
||||
bm := normalizeText(b.Manufacturer)
|
||||
return am != "" && am == bm
|
||||
}
|
||||
|
||||
func hasMACOverlap(a, b []string) bool {
|
||||
if len(a) == 0 || len(b) == 0 {
|
||||
return false
|
||||
}
|
||||
set := make(map[string]struct{}, len(a))
|
||||
for _, mac := range a {
|
||||
key := normalizeText(mac)
|
||||
if key != "" {
|
||||
set[key] = struct{}{}
|
||||
}
|
||||
}
|
||||
for _, mac := range b {
|
||||
if _, ok := set[normalizeText(mac)]; ok {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func sameKindFamily(a, b string) bool {
|
||||
if a == b {
|
||||
return true
|
||||
}
|
||||
family := map[string]bool{
|
||||
models.DeviceKindPCIe: true,
|
||||
models.DeviceKindGPU: true,
|
||||
models.DeviceKindNetwork: true,
|
||||
}
|
||||
return family[a] && family[b]
|
||||
}
|
||||
|
||||
func normalizeText(v string) string {
|
||||
s := strings.ToLower(strings.TrimSpace(v))
|
||||
s = strings.ReplaceAll(s, " ", "")
|
||||
s = strings.ReplaceAll(s, "_", "")
|
||||
s = strings.ReplaceAll(s, "-", "")
|
||||
return s
|
||||
}
|
||||
|
||||
func normalizeSlot(slot string) string {
|
||||
return normalizeText(slot)
|
||||
}
|
||||
|
||||
func qualityScore(d models.HardwareDevice) int {
|
||||
score := 0
|
||||
if normalizedSerial(d.SerialNumber) != "" {
|
||||
score += 6
|
||||
}
|
||||
if strings.TrimSpace(d.BDF) != "" {
|
||||
score += 4
|
||||
}
|
||||
if strings.TrimSpace(d.Model) != "" {
|
||||
score += 3
|
||||
}
|
||||
if strings.TrimSpace(d.Firmware) != "" {
|
||||
score += 2
|
||||
}
|
||||
if strings.TrimSpace(d.Status) != "" {
|
||||
score++
|
||||
}
|
||||
return score
|
||||
}
|
||||
|
||||
func normalizedSerial(serial string) string {
|
||||
s := strings.TrimSpace(serial)
|
||||
if s == "" {
|
||||
return ""
|
||||
}
|
||||
switch strings.ToUpper(s) {
|
||||
case "N/A", "NA", "NONE", "NULL", "UNKNOWN", "-":
|
||||
return ""
|
||||
default:
|
||||
return s
|
||||
}
|
||||
}
|
||||
|
||||
func buildFirmwareBySlot(firmware []models.FirmwareInfo) map[string]slotFirmwareInfo {
|
||||
out := make(map[string]slotFirmwareInfo)
|
||||
add := func(slot, model, version, category string) {
|
||||
key := normalizeSlotKey(slot)
|
||||
if key == "" || strings.TrimSpace(version) == "" {
|
||||
return
|
||||
}
|
||||
existing, ok := out[key]
|
||||
if ok && strings.TrimSpace(existing.Model) != "" {
|
||||
return
|
||||
}
|
||||
out[key] = slotFirmwareInfo{
|
||||
Model: strings.TrimSpace(model),
|
||||
Version: strings.TrimSpace(version),
|
||||
Category: category,
|
||||
}
|
||||
}
|
||||
|
||||
for _, fw := range firmware {
|
||||
name := strings.TrimSpace(fw.DeviceName)
|
||||
if name == "" {
|
||||
continue
|
||||
}
|
||||
if m := psuFirmwareRe.FindStringSubmatch(name); len(m) == 3 {
|
||||
model := strings.TrimSpace(m[2])
|
||||
if model == "" {
|
||||
model = "PSU"
|
||||
}
|
||||
add(m[1], model, fw.Version, "psu")
|
||||
continue
|
||||
}
|
||||
if m := nicFirmwareRe.FindStringSubmatch(name); len(m) == 3 {
|
||||
model := strings.TrimSpace(m[2])
|
||||
if model == "" {
|
||||
model = "NIC"
|
||||
}
|
||||
add(m[1], model, fw.Version, "nic")
|
||||
continue
|
||||
}
|
||||
if m := gpuFirmwareRe.FindStringSubmatch(name); len(m) == 3 {
|
||||
model := strings.TrimSpace(m[2])
|
||||
if model == "" {
|
||||
model = "GPU"
|
||||
}
|
||||
add(m[1], model, fw.Version, "gpu")
|
||||
continue
|
||||
}
|
||||
if m := nvsFirmwareRe.FindStringSubmatch(name); len(m) == 3 {
|
||||
model := strings.TrimSpace(m[2])
|
||||
if model == "" {
|
||||
model = "NVSwitch"
|
||||
}
|
||||
add(m[1], model, fw.Version, "nvswitch")
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
return out
|
||||
}
|
||||
|
||||
func normalizeSlotKey(slot string) string {
|
||||
return strings.ToLower(strings.TrimSpace(slot))
|
||||
}
|
||||
Reference in New Issue
Block a user