-
Notifications
You must be signed in to change notification settings - Fork 90
/
process.go
88 lines (77 loc) · 1.8 KB
/
process.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
//go:build linux
package process
import (
"context"
"time"
"github.com/go-kit/log"
"github.com/grafana/alloy/internal/component"
"github.com/grafana/alloy/internal/component/discovery"
"github.com/grafana/alloy/internal/featuregate"
)
func init() {
component.Register(component.Registration{
Name: "discovery.process",
Stability: featuregate.StabilityPublicPreview,
Args: Arguments{},
Exports: discovery.Exports{},
Build: func(opts component.Options, args component.Arguments) (component.Component, error) {
return New(opts, args.(Arguments))
},
})
}
func New(opts component.Options, args Arguments) (*Component, error) {
c := &Component{
l: opts.Logger,
onStateChange: opts.OnStateChange,
argsUpdates: make(chan Arguments),
args: args,
}
return c, nil
}
type Component struct {
l log.Logger
onStateChange func(e component.Exports)
processes []discovery.Target
argsUpdates chan Arguments
args Arguments
}
func (c *Component) Run(ctx context.Context) error {
doDiscover := func() error {
processes, err := discover(c.l, &c.args.DiscoverConfig)
if err != nil {
return err
}
c.processes = convertProcesses(processes)
c.changed()
return nil
}
if err := doDiscover(); err != nil {
return err
}
t := time.NewTicker(c.args.RefreshInterval)
defer t.Stop()
for {
select {
case <-ctx.Done():
return nil
case <-t.C:
if err := doDiscover(); err != nil {
return err
}
t.Reset(c.args.RefreshInterval)
case a := <-c.argsUpdates:
c.args = a
c.changed()
}
}
}
func (c *Component) Update(args component.Arguments) error {
a := args.(Arguments)
c.argsUpdates <- a
return nil
}
func (c *Component) changed() {
c.onStateChange(discovery.Exports{
Targets: join(c.processes, c.args.Join),
})
}