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

package kubemodel

// @bingen:generate:Resource
type Resource string

const (
	ResourceCPU          Resource = "cpu"
	ResourceMemory       Resource = "memory"
	ResourceNvidia       Resource = "nvidia.com/gpu"
	ResourceNvidiaShared Resource = "nvidia.com/gpu.shared"
	ResourceAWSVGPU      Resource = "k8s.amazonaws.com/vgpu"
)

// @bingen:generate:ResourceQuantity
type ResourceQuantity struct {
	Resource Resource `json:"resource"` // @bingen:field[version=1]
	Unit     Unit     `json:"unit"`     // @bingen:field[version=1]
	Values   Stats    `json:"values"`   // @bingen:field[version=1]
}

// @bingen:generate:ResourceQuantities
type ResourceQuantities map[Resource]ResourceQuantity

func (rqs ResourceQuantities) Set(resource Resource, unit Unit, statType StatType, value float64) {
	if _, ok := rqs[resource]; !ok {
		rqs[resource] = ResourceQuantity{
			Resource: resource,
			Unit:     unit,
			Values:   NewStats(),
		}
	}

	rqs[resource].Values[statType] = value
}