forked from grafana/loki
-
Notifications
You must be signed in to change notification settings - Fork 0
/
manager.go
114 lines (101 loc) · 2.65 KB
/
manager.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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package targets
import (
"github.com/go-kit/kit/log"
"github.com/pkg/errors"
"github.com/grafana/loki/pkg/promtail/api"
"github.com/grafana/loki/pkg/promtail/positions"
"github.com/grafana/loki/pkg/promtail/scrape"
)
type targetManager interface {
Ready() bool
Stop()
ActiveTargets() map[string][]Target
AllTargets() map[string][]Target
}
// TargetManagers manages a list of target managers.
type TargetManagers struct {
targetManagers []targetManager
}
// NewTargetManagers makes a new TargetManagers
func NewTargetManagers(
logger log.Logger,
positions *positions.Positions,
client api.EntryHandler,
scrapeConfigs []scrape.Config,
targetConfig *Config,
) (*TargetManagers, error) {
var targetManagers []targetManager
var fileScrapeConfigs []scrape.Config
var journalScrapeConfigs []scrape.Config
for _, cfg := range scrapeConfigs {
if cfg.HasServiceDiscoveryConfig() {
fileScrapeConfigs = append(fileScrapeConfigs, cfg)
}
}
if len(fileScrapeConfigs) > 0 {
fileTargetManager, err := NewFileTargetManager(
logger,
positions,
client,
fileScrapeConfigs,
targetConfig,
)
if err != nil {
return nil, errors.Wrap(err, "failed to make file target manager")
}
targetManagers = append(targetManagers, fileTargetManager)
}
for _, cfg := range scrapeConfigs {
if cfg.JournalConfig != nil {
journalScrapeConfigs = append(journalScrapeConfigs, cfg)
}
}
if len(journalScrapeConfigs) > 0 {
journalTargetManager, err := NewJournalTargetManager(
logger,
positions,
client,
journalScrapeConfigs,
)
if err != nil {
return nil, errors.Wrap(err, "failed to make journal target manager")
}
targetManagers = append(targetManagers, journalTargetManager)
}
return &TargetManagers{targetManagers: targetManagers}, nil
}
// ActiveTargets returns active targets per jobs
func (tm *TargetManagers) ActiveTargets() map[string][]Target {
result := map[string][]Target{}
for _, t := range tm.targetManagers {
for job, targets := range t.ActiveTargets() {
result[job] = append(result[job], targets...)
}
}
return result
}
// AllTargets returns all targets per jobs
func (tm *TargetManagers) AllTargets() map[string][]Target {
result := map[string][]Target{}
for _, t := range tm.targetManagers {
for job, targets := range t.AllTargets() {
result[job] = append(result[job], targets...)
}
}
return result
}
// Ready if there's at least one ready target manager.
func (tm *TargetManagers) Ready() bool {
for _, t := range tm.targetManagers {
if t.Ready() {
return true
}
}
return false
}
// Stop the TargetManagers.
func (tm *TargetManagers) Stop() {
for _, t := range tm.targetManagers {
t.Stop()
}
}