package historian

import (
	"context"
	"errors"

	"github.com/grafana/grafana-plugin-sdk-go/data"
	ngmodels "github.com/grafana/grafana/pkg/services/ngalert/models"
	"github.com/grafana/grafana/pkg/services/ngalert/state"
	history_model "github.com/grafana/grafana/pkg/services/ngalert/state/historian/model"
)

type Backend interface {
	Record(ctx context.Context, rule history_model.RuleMeta, states []state.StateTransition) <-chan error
	Query(ctx context.Context, query ngmodels.HistoryQuery) (*data.Frame, error)
}

// MultipleBackend is a state.Historian that records history to multiple backends at once.
// Only one backend is used for reads. The backend selected for read traffic is called the primary and all others are called secondaries.
type MultipleBackend struct {
	primary     Backend
	secondaries []Backend
}

func NewMultipleBackend(primary Backend, secondaries ...Backend) *MultipleBackend {
	return &MultipleBackend{
		primary:     primary,
		secondaries: secondaries,
	}
}

func (h *MultipleBackend) Record(ctx context.Context, rule history_model.RuleMeta, states []state.StateTransition) <-chan error {
	jobs := make([]<-chan error, 0, len(h.secondaries)+1) // One extra for the primary.
	for _, b := range append([]Backend{h.primary}, h.secondaries...) {
		jobs = append(jobs, b.Record(ctx, rule, states))
	}
	errCh := make(chan error, 1)
	go func() {
		defer close(errCh)
		errs := make([]error, 0)
		// Wait for all jobs to complete. Order doesn't matter here, as we always need to wait on the slowest job regardless.
		for _, ch := range jobs {
			err := <-ch
			if err != nil {
				errs = append(errs, err)
			}
		}
		errCh <- errors.Join(errs...)
	}()
	return errCh
}

func (h *MultipleBackend) Query(ctx context.Context, query ngmodels.HistoryQuery) (*data.Frame, error) {
	return h.primary.Query(ctx, query)
}