-
Notifications
You must be signed in to change notification settings - Fork 0
/
targetmanager.go
79 lines (67 loc) · 1.92 KB
/
targetmanager.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package gcplog
import (
"fmt"
"github.com/go-kit/kit/log"
"github.com/go-kit/kit/log/level"
"github.com/metrico/loki-apache/pkg/logentry/stages"
"github.com/metrico/loki-apache/pkg/promtail/api"
"github.com/metrico/loki-apache/pkg/promtail/scrapeconfig"
"github.com/metrico/loki-apache/pkg/promtail/targets/target"
)
// nolint: golint
type GcplogTargetManager struct {
logger log.Logger
targets map[string]*GcplogTarget
}
func NewGcplogTargetManager(
metrics *Metrics,
logger log.Logger,
client api.EntryHandler,
scrape []scrapeconfig.Config,
) (*GcplogTargetManager, error) {
tm := &GcplogTargetManager{
logger: logger,
targets: make(map[string]*GcplogTarget),
}
for _, cf := range scrape {
if cf.GcplogConfig == nil {
continue
}
pipeline, err := stages.NewPipeline(log.With(logger, "component", "pubsub_pipeline"), cf.PipelineStages, &cf.JobName, metrics.reg)
if err != nil {
return nil, err
}
t, err := NewGcplogTarget(metrics, logger, pipeline.Wrap(client), cf.RelabelConfigs, cf.JobName, cf.GcplogConfig)
if err != nil {
return nil, fmt.Errorf("failed to create pubsub target: %w", err)
}
tm.targets[cf.JobName] = t
}
return tm, nil
}
func (tm *GcplogTargetManager) Ready() bool {
for _, t := range tm.targets {
if t.Ready() {
return true
}
}
return false
}
func (tm *GcplogTargetManager) Stop() {
for name, t := range tm.targets {
if err := t.Stop(); err != nil {
level.Error(t.logger).Log("event", "failed to stop pubsub target", "name", name, "cause", err)
}
}
}
func (tm *GcplogTargetManager) ActiveTargets() map[string][]target.Target {
// TODO(kavi): if someway to check if specific topic is active and store the state on the target struct?
return tm.AllTargets()
}
func (tm *GcplogTargetManager) AllTargets() map[string][]target.Target {
res := make(map[string][]target.Target, len(tm.targets))
for k, v := range tm.targets {
res[k] = []target.Target{v}
}
return res
}