Alexander Akhmetov
56 lines
1.8 KiB
Go
56 lines
1.8 KiB
Go
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)
|
|
}
|