-
Notifications
You must be signed in to change notification settings - Fork 2
/
upbot.go
102 lines (89 loc) · 2.05 KB
/
upbot.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
// Copyright 2020 Changkun Ou. All rights reserved.
// Use of this source code is governed by a MIT
// license that can be found in the LICENSE file.
package main
import (
"context"
"log"
"net/http"
"os"
"os/signal"
"sync"
"time"
)
func main() {
log.SetPrefix("upbot: ")
log.SetFlags(log.LstdFlags | log.Lshortfile | log.Lmsgprefix)
ctx, cancel := context.WithCancel(context.Background())
go func() {
q := make(chan os.Signal, 1)
signal.Notify(q, os.Interrupt, os.Kill)
sig := <-q
log.Printf("%v", sig)
cancel()
}()
group := map[int][]*monitor{}
group[conf.Default.Interval] = []*monitor{}
for _, m := range conf.Monitors {
i := conf.Default.Interval
if m.Interval != 0 {
i = m.Interval
}
m.Status = statusStart
if _, ok := group[i]; ok {
group[i] = append(group[i], m)
} else {
group[i] = []*monitor{m}
}
}
go serve(ctx)
wg := sync.WaitGroup{}
wg.Add(len(group))
for interval, ms := range group {
_in := interval
_ms := ms
go func() {
defer wg.Done()
checker(ctx, _in, _ms)
}()
}
wg.Wait()
}
func serve(ctx context.Context) {
addr := os.Getenv("UPBOT_ADDR")
if len(addr) == 0 {
addr = conf.Addr
}
http.Handle(wrapper(app, "/upbot", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("I'm OK"))
})))
log.Printf("upbot is on: %s", addr)
if err := http.ListenAndServe(addr, nil); err != nil {
log.Fatalf("failed to start upbot: %v", err)
}
}
func checker(ctx context.Context, interval int, ms []*monitor) {
t := time.NewTicker(time.Second * time.Duration(interval))
for {
select {
case <-ctx.Done():
log.Println("upbot is down, goodbye!")
return
case <-t.C:
log.Println("checking...")
wg := sync.WaitGroup{}
for _, m := range ms {
go func(m *monitor) {
notifiers := []string{}
notifiers = append(notifiers, conf.Default.Notifiers...)
notifiers = append(notifiers, m.Notifiers...)
err := m.CheckHealth(ctx, notifiers)
if err != nil {
log.Printf("%s check err: %v", m.Name, err)
}
}(m)
}
wg.Wait()
}
}
}