/
journaltargetmanager_linux.go
96 lines (83 loc) · 2.38 KB
/
journaltargetmanager_linux.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
// +build cgo
package targets
import (
"github.com/go-kit/kit/log"
"github.com/go-kit/kit/log/level"
"github.com/grafana/loki/pkg/logentry/stages"
"github.com/grafana/loki/pkg/promtail/api"
"github.com/grafana/loki/pkg/promtail/positions"
"github.com/grafana/loki/pkg/promtail/scrape"
"github.com/prometheus/client_golang/prometheus"
)
// JournalTargetManager manages a series of JournalTargets.
type JournalTargetManager struct {
logger log.Logger
targets map[string]*JournalTarget
}
// NewJournalTargetManager creates a new JournalTargetManager.
func NewJournalTargetManager(
logger log.Logger,
positions *positions.Positions,
client api.EntryHandler,
scrapeConfigs []scrape.Config,
) (*JournalTargetManager, error) {
tm := &JournalTargetManager{
logger: logger,
targets: make(map[string]*JournalTarget),
}
for _, cfg := range scrapeConfigs {
if cfg.JournalConfig == nil {
continue
}
registerer := prometheus.DefaultRegisterer
pipeline, err := stages.NewPipeline(log.With(logger, "component", "journal_pipeline"), cfg.PipelineStages, &cfg.JobName, registerer)
if err != nil {
return nil, err
}
t, err := NewJournalTarget(
logger,
pipeline.Wrap(client),
positions,
cfg.JobName,
cfg.RelabelConfigs,
cfg.JournalConfig,
)
if err != nil {
return nil, err
}
tm.targets[cfg.JobName] = t
}
return tm, nil
}
// Ready returns true if at least one JournalTarget is also ready.
func (tm *JournalTargetManager) Ready() bool {
for _, t := range tm.targets {
if t.Ready() {
return true
}
}
return false
}
// Stop stops the JournalTargetManager and all of its JournalTargets.
func (tm *JournalTargetManager) Stop() {
for _, t := range tm.targets {
if err := t.Stop(); err != nil {
level.Error(t.logger).Log("msg", "error stopping JournalTarget", "err", err.Error())
}
}
}
// ActiveTargets returns the list of JournalTargets where journal data
// is being read. ActiveTargets is an alias to AllTargets as
// JournalTargets cannot be deactivated, only stopped.
func (tm *JournalTargetManager) ActiveTargets() map[string][]Target {
return tm.AllTargets()
}
// AllTargets returns the list of all targets where journal data
// is currently being read.
func (tm *JournalTargetManager) AllTargets() map[string][]Target {
result := make(map[string][]Target, len(tm.targets))
for k, v := range tm.targets {
result[k] = []Target{v}
}
return result
}