forked from rancher/container-crontab
/
handler.go
100 lines (84 loc) · 2.58 KB
/
handler.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
package events
import (
"context"
"github.com/Sirupsen/logrus"
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/events"
"github.com/docker/docker/client"
"github.com/prometheus/client_golang/prometheus"
"github.com/rancher/container-crontab/cron"
)
// Handler handles messages
type Handler interface {
Handle(Message)
}
// Message is a message from an event stream
type Message *events.Message
// DockerHandler handles docker messages
type DockerHandler struct {
Crontab *cron.Crontab
}
type DockerHandlerOpts struct {
RancherMode bool
MetadataURL string
}
// NewDockerHandler returns a docker handler with crontab
func NewDockerHandler(opts *DockerHandlerOpts) (*DockerHandler, error) {
crontab, err := cron.NewCrontab()
if err != nil {
return nil, err
}
if opts.RancherMode {
logrus.Infof("Using Rancher Mode with metadata URL = %s", opts.MetadataURL)
crontab, err = cron.NewRancherTypeCrontab(opts.MetadataURL)
if err != nil {
return nil, err
}
}
dClient, err := client.NewEnvClient()
if err != nil {
return nil, err
}
defer dClient.Close()
containers, err := dClient.ContainerList(context.Background(), types.ContainerListOptions{
All: true,
})
if err != nil {
logrus.Fatal(err)
return nil, err
}
// Scan containers
logrus.Infof("Scanning for container cron entries")
for _, container := range containers {
if _, ok := container.Labels["cron.schedule"]; ok {
crontab.AddJob(container.ID, container.Labels, "docker")
}
}
return &DockerHandler{
Crontab: crontab,
}, nil
}
// Handle implements handler interface
func (dh DockerHandler) Handle(msg Message) {
// Adding a cron.schedule label flags the container for deeper inspection
// With this service
if _, ok := msg.Actor.Attributes["cron.schedule"]; ok {
if msg.Action == "start" || msg.Action == "create" {
logrus.Debugf("Processing %s event for container: %s", msg.Action, msg.ID)
dh.Crontab.AddJob(msg.ID, msg.Actor.Attributes, "docker")
}
if msg.Action == "stop" || msg.Action == "die" {
logrus.Debugf("Proccessing %s event for container: %s", msg.Action, msg.ID)
dh.Crontab.DeactivateJob(msg.ID, msg.Actor.Attributes)
}
if msg.Action == "destroy" {
logrus.Debugf("Processing destroy event for container: %s", msg.ID)
dh.Crontab.RemoveJob(msg.ID)
}
}
}
func (dh DockerHandler) GetJobStats(guage *prometheus.GaugeVec) (*prometheus.GaugeVec, error) {
guage.With(prometheus.Labels{"state": "active"}).Set(dh.Crontab.GetNumberOfActiveJobs())
guage.With(prometheus.Labels{"state": "inactive"}).Set(dh.Crontab.GetNumberOfInactiveJobs())
return guage, nil
}