cost-model/core/pkg/model/kubemodel/dcgm.go

package kubemodel

import (
	"fmt"
	"time"
)

// DCGMDevice holds recording from the DCGM exporter which provides identification and usage metrics for
// Nvidia gpu. These Nvidia devices can be incorporated into the cluster via k8s Device Plugin API or DRAs.
// While the DCGM exporter does provide unique identifiers for the containers that it is reporting metrics on,
// It is split out here to provide some isolate from the rest of the KubeModel which represent universal structures
// from the k8s API. It is left to the end user to interpret the relationships to the rest of the cluster based on
// container unique identifiers
// @bingen:generate:DCGMDevice
type DCGMDevice struct {
	UUID      string             `json:"uuid"`
	Start     time.Time          `json:"start"`
	End       time.Time          `json:"end"`
	Device    string             `json:"device"`
	ModelName string             `json:"modelName"`
	PodUsages map[string]DCGMPod `json:"podUsages"`
}

// @bingen:generate:DCGMPod
type DCGMPod struct {
	ContainerUsages map[string]DCGMContainer `json:"container-usages"`
}

// @bingen:generate:DCGMContainer
type DCGMContainer struct {
	UsageAvg float64 `json:"usageAvg"`
	UsageMax float64 `json:"usageMax"`
}

func (d *DCGMDevice) ValidateDCGMDevice(window Window) error {
	if d.UUID == "" {
		return fmt.Errorf("UUID is missing for DCGMDevice with device '%s'", d.Device)
	}

	if err := checkWindow(window, d.Start, d.End); err != nil {
		return err
	}

	return nil
}

// RegisterDCGMDevice validates and adds a DCGMDevice to the set, keyed by UUID.
func (kms *KubeModelSet) RegisterDCGMDevice(device *DCGMDevice) error {
	if err := device.ValidateDCGMDevice(kms.Window); err != nil {
		err = fmt.Errorf("RegisterDCGMDevice: invalid dcgm device: %w", err)
		kms.Error(err)
		return err
	}

	if _, ok := kms.DCGMDevices[device.UUID]; !ok {
		if kms.Cluster == nil {
			kms.Warnf("RegisterDCGMDevice: Cluster is nil")
		}

		kms.DCGMDevices[device.UUID] = device

		kms.Metadata.ObjectCount++
	}

	return nil
}