-
Notifications
You must be signed in to change notification settings - Fork 0
/
component.go
55 lines (50 loc) · 1.44 KB
/
component.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
package monitor
import (
"github.com/D3rus/go-library/app"
"github.com/D3rus/go-library/configurator"
"github.com/D3rus/go-library/container"
"github.com/D3rus/go-library/logger"
"github.com/D3rus/go-library/runner"
"github.com/D3rus/go-library/socket/client"
"github.com/D3rus/go-library/socket/client/connector"
"github.com/D3rus/go-library/socket/client/monitor/counter"
"github.com/spf13/pflag"
)
func BindFlags(flagSet *pflag.FlagSet, container container.Container) error {
if err := connector.BindFlags(flagSet, container); err != nil {
return err
}
return container.Invoke(func(config *Config) {
flagSet.DurationVar(
&config.DelayReconnect,
DelayReconnectFieldName,
DelayReconnectDefault,
"timeout between reconnect attempts",
)
})
}
var Component = &app.Component{
Dependencies: app.Components{
logger.Component,
configurator.Component,
runner.Component,
counter.Component,
},
Constructor: app.Constructor(func(container container.Container) error {
return container.Provides(
connector.NewConfig,
NewConfig,
NewWithConfigurator,
func(client Monitor) client.Client { return client },
)
}),
BindFlags: app.BindFlags(BindFlags),
Run: app.Run(func(container container.Container) error {
if err := container.Invoke(connector.Configuration); err != nil {
return err
}
return container.Invoke(func(client Monitor, runner runner.Runner) error {
return runner.AddProcesses(client)
})
}),
}