summaryrefslogtreecommitdiff
path: root/main.go
blob: de89840221741d372a23a9e2f30e1757867f2ffe (plain)
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
package main

import (
	"flag"
	"fmt"
	"log"
	"sync"
	"time"
)

var (
	conf = flag.String("conf", "config.yml", "configuration file")
	intr = flag.Duration("interval", time.Second*5, "check interval")
)

func Checker(uri string, intr time.Duration, errc chan<- error) {
	t := time.NewTicker(intr)
	var stalled bool
	defer t.Stop()
	key := Keys()
	for range t.C {
		err := Check(uri, <-key)
		if err != nil {
			stalled = true
			errc <- err
		} else if stalled {
			stalled = false
			errc <- fmt.Errorf("%v back online", uri)
		}
	}
}

func main() {
	flag.Parse()
	c, err := LoadConfig(*conf)
	if err != nil {
		log.Fatal(err)
	}
	errc := Logger()
	wg := sync.WaitGroup{}
	for _, uri := range c.Check {
		wg.Add(1)
		go func(uri string) {
			Checker(uri, *intr, errc)
			wg.Done()
		}(uri)
	}
	wg.Wait()
}