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

package kubemodel

import (
	"fmt"
	"time"
)

// @bingen:generate:Container
type Container struct {
	PodUID           string             `json:"podUid"`
	Name             string             `json:"name"`
	ResourceRequests ResourceQuantities `json:"resourceRequests"`
	ResourceLimits   ResourceQuantities `json:"resourceLimits"`
	CPUCoreUsageAvg  float64            `json:"cpuCoreUsageAvg"`
	CPUCoreUsageMax  float64            `json:"cpuCoreUsageMax"`
	RAMBytesUsageAvg float64            `json:"ramBytesUsageAvg"`
	RAMBytesUsageMax float64            `json:"ramBytesUsageMax"`
	Start            time.Time          `json:"start"`
	End              time.Time          `json:"end"`
}

func (c *Container) GetKey() string {
	return fmt.Sprintf("%s/%s", c.PodUID, c.Name)
}

func (c *Container) ValidateContainer(window Window) error {
	if c.PodUID == "" {
		return fmt.Errorf("PodUID is missing for Container with name '%s'", c.Name)
	}

	if c.Name == "" {
		return fmt.Errorf("Name is missing for Container on pod '%s'", c.PodUID)
	}

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

	return nil
}

func (kms *KubeModelSet) RegisterContainer(container *Container) error {
	if err := container.ValidateContainer(kms.Window); err != nil {
		err = fmt.Errorf("RegisterContainer: invalid container: %w", err)
		kms.Error(err)
		return err
	}

	key := container.GetKey()
	if _, ok := kms.Containers[key]; !ok {
		kms.Containers[key] = container
		kms.Metadata.ObjectCount++
	}

	return nil
}