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"`
	CPUCoresAllocated float64            `json:"cpuCoresAllocated"`
	CPUCoreUsageAvg   float64            `json:"cpuCoreUsageAvg"`
	CPUCoreUsageMax   float64            `json:"cpuCoreUsageMax"`
	RAMBytesAllocated float64            `json:"ramBytesAllocated"`
	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
}