cloud/rca.go

Rama main · réplica de lectura del repositorio original.

package cloud

import (
	"bytes"
	"context"
	"net/http"

	lz4 "github.com/DataDog/golz4"
	"github.com/coroot/coroot/db"
	"github.com/coroot/coroot/model"
	"github.com/coroot/coroot/timeseries"
	"github.com/vmihailenco/msgpack/v5"
)

type RCARequest struct {
	Ctx timeseries.Context

	ApplicationId model.ApplicationId

	CheckConfigs                model.CheckConfigs
	ApplicationDeployments      map[model.ApplicationId][]*model.ApplicationDeployment
	ApplicationCategorySettings map[model.ApplicationCategory]*db.ApplicationCategorySettings
	CustomApplications          map[string]model.CustomApplication
	CustomCloudPricing          *db.CustomCloudPricing

	Metrics map[string][]*model.MetricValues

	KubernetesEvents []*model.LogEntry

	ErrorTrace *model.Trace
	SlowTrace  *model.Trace
}

func (api *Api) RCA(ctx context.Context, req RCARequest) (*model.RCA, error) {
	buf := bytes.NewBuffer(nil)
	lw := lz4.NewWriter(buf)
	if err := msgpack.NewEncoder(lw).Encode(req); err != nil {
		return nil, err
	}
	if err := lw.Close(); err != nil {
		return nil, err
	}

	var rca model.RCA
	if err := api.request(ctx, http.MethodPost, "/integration/rca", "application/msgpack", "lz4", buf, &rca); err != nil {
		return nil, err
	}
	return &rca, nil
}

func (api *Api) RCAStatus(ctx context.Context, incidentsAutoInvestigation bool) (string, error) {
	settings, err := api.GetSettings()
	if err != nil {
		return "Failed", err
	}
	if settings.ApiKey == "" {
		return "AI disabled", nil
	}
	if incidentsAutoInvestigation && settings.RCA.DisableIncidentsAutoInvestigation {
		return "AI disabled", nil
	}
	info, err := api.IntegrationInfo(ctx)
	if err != nil {
		return "Failed", err
	}
	if info.RCA == nil {
		return "AI disabled", nil
	}
	if info.RCA.CreditsSpent >= info.RCA.CreditsTotal {
		return "Out of credits", err
	}
	return "OK", nil
}