-
Notifications
You must be signed in to change notification settings - Fork 2k
/
multitsdb.go
41 lines (34 loc) · 1.24 KB
/
multitsdb.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// Copyright (c) The Thanos Authors.
// Licensed under the Apache License 2.0.
package exemplars
import (
"github.com/pkg/errors"
"github.com/prometheus/prometheus/promql/parser"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"github.com/thanos-io/thanos/pkg/exemplars/exemplarspb"
)
// MultiTSDB implements exemplarspb.ExemplarsServer that allows to fetch exemplars a MultiTSDB instance.
type MultiTSDB struct {
tsdbExemplarsServers func() map[string]*TSDB
}
// NewMultiTSDB creates new exemplars.MultiTSDB.
func NewMultiTSDB(tsdbExemplarsServers func() map[string]*TSDB) *MultiTSDB {
return &MultiTSDB{
tsdbExemplarsServers: tsdbExemplarsServers,
}
}
// Exemplars returns all specified exemplars from a MultiTSDB instance.
func (m *MultiTSDB) Exemplars(r *exemplarspb.ExemplarsRequest, s exemplarspb.Exemplars_ExemplarsServer) error {
expr, err := parser.ParseExpr(r.Query)
if err != nil {
return status.Error(codes.Internal, err.Error())
}
matchers := parser.ExtractSelectors(expr)
for tenant, es := range m.tsdbExemplarsServers() {
if err := es.Exemplars(matchers, r.Start, r.End, s); err != nil {
return status.Error(codes.Aborted, errors.Wrapf(err, "get exemplars for tenant %s", tenant).Error())
}
}
return nil
}