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

package collector

import (
	coreenv "github.com/opencost/opencost/core/pkg/env"
	"github.com/opencost/opencost/modules/collector-source/pkg/env"
	"github.com/opencost/opencost/modules/collector-source/pkg/util"
)

type CollectorConfig struct {
	Resolutions    []util.ResolutionConfiguration `json:"resolutions"`
	ScrapeInterval string                         `json:"scrape_interval"`
	ClusterID      string                         `json:"cluster_id"`
	NetworkPort    int                            `json:"network_port"`
}

func NewOpenCostCollectorConfigFromEnv() CollectorConfig {
	return CollectorConfig{
		Resolutions: []util.ResolutionConfiguration{
			{
				Interval:  "10m",
				Retention: env.GetCollector10mResolutionRetention(),
			},
			{
				Interval:  "1h",
				Retention: env.GetCollector1hResolutionRetention(),
			},
			{
				Interval:  "1d",
				Retention: env.GetCollection1dResolutionRetention(),
			},
		},
		ScrapeInterval: env.GetCollectorScrapeIntervalSeconds(),
		ClusterID:      coreenv.GetClusterID(),
		NetworkPort:    env.GetNetworkPort(),
	}
}