cost-model/core/pkg/opencost/exporter/allocation/source.go

package allocation

import (
	"time"

	"github.com/opencost/opencost/core/pkg/exporter"
	"github.com/opencost/opencost/core/pkg/opencost"
	"github.com/opencost/opencost/core/pkg/pipelines"
)

type AllocationSource interface {
	ComputeAllocation(start, end time.Time, resolution time.Duration) (*opencost.AllocationSet, error)
}

type AllocationComputeSource struct {
	src AllocationSource
}

// NewAllocationComputeSource creates an `exporter.ComputeSource[opencost.AllocationSet]` implementation
func NewAllocationComputeSource(src AllocationSource) exporter.ComputeSource[opencost.AllocationSet] {
	return &AllocationComputeSource{
		src: src,
	}
}

// CanCompute should return true iff the ComputeSource can effectively act as
// a source of T data for the given time range. For example, a ComputeSource
// with two-day coverage cannot fulfill a range from three days ago, and should
// not be left to return an error in Compute. Instead, it should report that is
// cannot compute and allow another Source to handle the computation.
func (acs *AllocationComputeSource) CanCompute(start, end time.Time) bool {
	return true
}

// Compute should compute a single T for the given time range, optionally using the given resolution.
func (acs *AllocationComputeSource) Compute(start, end time.Time, resolution time.Duration) (*opencost.AllocationSet, error) {
	return acs.src.ComputeAllocation(start, end, resolution)
}

// Name returns the name of the ComputeSource
func (acs *AllocationComputeSource) Name() string {
	return pipelines.AllocationPipelineName
}