cost-model/core/pkg/diagnostics/exporter/source.go

package exporter

import (
	"context"
	"time"

	"github.com/opencost/opencost/core/pkg/diagnostics"
)

// DiagnosticSource is an `export.ExportSource` implementation that provides the basic data for a `DiagnosticResult` payload.
type DiagnosticSource struct {
	applicationName   string
	diagnosticService diagnostics.DiagnosticService
}

// NewDiagnosticSource creates a new `DiagnosticSource` instance. It accepts the `DiagnosticService` implementation
// that will be used to retrieve the diagnostic results.
func NewDiagnosticSource(applicationName string, diagnosticService diagnostics.DiagnosticService) *DiagnosticSource {
	return &DiagnosticSource{
		applicationName:   applicationName,
		diagnosticService: diagnosticService,
	}
}

// Make creates a new `DiagnosticsRunReport` instance with the provided current time.
func (ds *DiagnosticSource) Make(t time.Time) *diagnostics.DiagnosticsRunReport {
	ctx := context.Background()

	// returning nil will prevent export -- skip for 0 registered diagnostics
	if ds.diagnosticService.Total() == 0 {
		return nil
	}

	return &diagnostics.DiagnosticsRunReport{
		StartTime:   t,
		Application: ds.applicationName,
		Results:     ds.diagnosticService.Run(ctx),
	}
}

func (ds *DiagnosticSource) Name() string {
	return diagnostics.DiagnosticsEventName + "-source"
}