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

package collector

import (
	"fmt"

	"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"`
	BucketConfigFile string                         `json:"bucket_config_file"`
}

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