cost-model/modules/collector-source/pkg/metric/collector.go

package metric

import (
	"maps"
	"sort"
	"time"

	"github.com/opencost/opencost/modules/collector-source/pkg/metric/aggregator"
	"github.com/opencost/opencost/modules/collector-source/pkg/util"
)

// MetricCollectorID is a unique identifier for a specific metric collector instance. We
// use this identifier to register and unregister metric instances from the metrics metric
// instead of the metric name and aggregation type to allow selectable cardinality (via Labels)
// across multiple instances of the same aggregation type and metric name.
type MetricCollectorID string

const (
	PVPricePerGiBHourID             MetricCollectorID = "PVPricePerGiBHour"
	PVUsedAverageID                 MetricCollectorID = "PVUsedAverage"
	PVUsedMaxID                     MetricCollectorID = "PVUsedMax"
	PVCInfoID                       MetricCollectorID = "PVCInfo"
	PVActiveMinutesID               MetricCollectorID = "PVActiveMinutes"
	LocalStorageUsedActiveMinutesID MetricCollectorID = "LocalStorageUsedCost"
	LocalStorageUsedAverageID       MetricCollectorID = "LocalStorageUsedAverage"
	LocalStorageUsedMaxID           MetricCollectorID = "LocalStorageUsedMax"
	LocalStorageBytesID             MetricCollectorID = "LocalStorageBytesID"
	LocalStorageActiveMinutesID     MetricCollectorID = "LocalStorageActiveMinutes"
	NodeCPUCoresCapacityID          MetricCollectorID = "NodeCPUCoresCapacity"
	NodeCPUCoresAllocatableID       MetricCollectorID = "NodeCPUCoresAllocatable"
	NodeRAMBytesCapacityID          MetricCollectorID = "NodeRAMBytesCapacity"
	NodeRAMBytesAllocatableID       MetricCollectorID = "NodeRAMBytesAllocatable"
	NodeGPUCountID                  MetricCollectorID = "NodeGPUCount"
	NodeLabelsID                    MetricCollectorID = "NodeLabels"
	NodeActiveMinutesID             MetricCollectorID = "NodeActiveMinutes"
	NodeCPUModeTotalID              MetricCollectorID = "NodeCPUModeTotal"
	NodeRAMSystemUsageAverageID     MetricCollectorID = "NodeRAMSystemUsageAverage"
	NodeRAMUserUsageAverageID       MetricCollectorID = "NodeRAMUserUsageAverage"
	LBPricePerHourID                MetricCollectorID = "LBPricePerHour"
	LBActiveMinutesID               MetricCollectorID = "LBActiveMinutes"
	ClusterManagementDurationID     MetricCollectorID = "ClusterManagementDuration"
	ClusterManagementPricePerHourID MetricCollectorID = "ClusterManagementPricePerHour"
	PodActiveMinutesID              MetricCollectorID = "PodActiveMinutes"
	RAMBytesAllocatedID             MetricCollectorID = "RAMBytesAllocated"
	RAMRequestsID                   MetricCollectorID = "RAMRequests"
	RAMUsageAverageID               MetricCollectorID = "RAMUsageAverage"
	RAMUsageMaxID                   MetricCollectorID = "RAMUsageMax"
	CPUCoresAllocatedID             MetricCollectorID = "CPUCoresAllocated"
	CPURequestsID                   MetricCollectorID = "CPURequestsID"
	CPUUsageAverageID               MetricCollectorID = "CPUUsageAverage"
	CPUUsageMaxID                   MetricCollectorID = "CPUUsageMax"
	GPUsRequestedID                 MetricCollectorID = "GPUsRequested"
	GPUsUsageAverageID              MetricCollectorID = "GPUsUsageAverage"
	GPUsUsageMaxID                  MetricCollectorID = "GPUsUsageMax"
	GPUsAllocatedID                 MetricCollectorID = "GPUsAllocated"
	IsGPUSharedID                   MetricCollectorID = "IsGPUShared"
	GPUInfoID                       MetricCollectorID = "GPUInfo"
	NodeCPUPricePerHourID           MetricCollectorID = "NodeCPUPricePerHour"
	NodeRAMPricePerGiBHourID        MetricCollectorID = "NodeRAMPricePerGiBHour"
	NodeGPUPricePerHourID           MetricCollectorID = "NodeGPUPricePerHour"
	NodeIsSpotID                    MetricCollectorID = "NodeIsSpot"
	PodPVCAllocationID              MetricCollectorID = "PodPVCAllocation"
	PVCBytesRequestedID             MetricCollectorID = "PVCBytesRequested"
	PVBytesID                       MetricCollectorID = "PVBytesID"
	PVInfoID                        MetricCollectorID = "PVInfo"
	NetZoneGiBID                    MetricCollectorID = "NetZoneGiB"
	NetZonePricePerGiBID            MetricCollectorID = "NetZonePricePerGiB"
	NetRegionGiBID                  MetricCollectorID = "NetRegionGiB"
	NetRegionPricePerGiBID          MetricCollectorID = "NetRegionPricePerGiB"
	NetInternetGiBID                MetricCollectorID = "NetInternetGiB"
	NetInternetPricePerGiBID        MetricCollectorID = "NetInternetPricePerGiB"
	NetInternetServiceGiBID         MetricCollectorID = "NetInternetServiceGiB"
	NetTransferBytesID              MetricCollectorID = "NetTransferBytes"
	NetZoneIngressGiBID             MetricCollectorID = "NetZoneIngressGiB"
	NetRegionIngressGiBID           MetricCollectorID = "NetRegionIngressGiB"
	NetInternetIngressGiBID         MetricCollectorID = "NetInternetIngressGiB"
	NetInternetServiceIngressGiBID  MetricCollectorID = "NetInternetServiceIngressGiB"
	NetReceiveBytesID               MetricCollectorID = "NetReceiveBytes"
	NamespaceLabelsID               MetricCollectorID = "NamespaceLabels"
	NamespaceAnnotationsID          MetricCollectorID = "NamespaceAnnotations"
	PodLabelsID                     MetricCollectorID = "PodLabels"
	PodAnnotationsID                MetricCollectorID = "PodAnnotations"
	ServiceLabelsID                 MetricCollectorID = "ServiceLabels"
	DeploymentLabelsID              MetricCollectorID = "DeploymentLabels"
	StatefulSetLabelsID             MetricCollectorID = "StatefulSetLabels"
	DaemonSetLabelsID               MetricCollectorID = "DaemonSetLabels"
	JobLabelsID                     MetricCollectorID = "JobLabels"
	PodsWithReplicaSetOwnerID       MetricCollectorID = "PodsWithReplicaSetOwner"
	ReplicaSetsWithoutOwnersID      MetricCollectorID = "ReplicaSetsWithoutOwners"
	ReplicaSetsWithRolloutID        MetricCollectorID = "ReplicaSetsWithRollout"
)

// MetricCollector is a data structure that represents a specific MetricCollector metric instance that contains its own breakdown
// of stored metrics by a specific label set.
type MetricCollector struct {
	id                MetricCollectorID // ie: RAMUsageAverage
	metricName        string            // ie: container_memory_working_set_bytes
	labels            []string
	aggregatorFactory aggregator.MetricAggregatorFactory
	metrics           map[uint64]aggregator.MetricAggregator // map[Hash(labelValues)] = aggregator
	filter            func(map[string]string) bool
}

// NewMetricCollector creates a new MetricCollector instance with a unique identifier. The metric name is the specific
// name of the collected metric that will be used to query the
func NewMetricCollector(id MetricCollectorID, metricName string, labels []string, aggregatorFactory aggregator.MetricAggregatorFactory, fn func(map[string]string) bool) *MetricCollector {
	return &MetricCollector{
		id:                id,
		metricName:        metricName,
		labels:            labels,
		aggregatorFactory: aggregatorFactory,
		metrics:           make(map[uint64]aggregator.MetricAggregator),
		filter:            fn,
	}
}

func (mi *MetricCollector) Update(labels map[string]string, value float64, timestamp time.Time, additionalInfo map[string]string) {
	if mi.filter != nil && !mi.filter(labels) {
		return
	}

	labelValues := make([]string, len(mi.labels))
	for i, key := range mi.labels {
		labelValues[i] = labels[key]
	}
	key := util.Hash(labelValues)
	if mi.metrics[key] == nil {
		mi.metrics[key] = mi.aggregatorFactory(
			util.MetricNameFor(mi.metricName, mi.labels, labelValues), labelValues)
	}

	mi.metrics[key].Update(value, timestamp, additionalInfo)
}

func (mi *MetricCollector) Get() []*aggregator.MetricResult {
	results := make([]*aggregator.MetricResult, 0, len(mi.metrics))
	for _, metric := range mi.metrics {
		labels := util.ToMap(mi.labels, metric.LabelValues())
		maps.Copy(labels, metric.AdditionInfo())
		mr := &aggregator.MetricResult{
			Name:         metric.Name(),
			MetricLabels: labels,
			Values:       metric.Value(),
		}

		results = append(results, mr)
	}

	sort.Slice(results, func(i, j int) bool {
		return results[i].Name < results[j].Name
	})
	
	return results
}

func (mi *MetricCollector) Labels() []string {
	return mi.labels
}