-
Notifications
You must be signed in to change notification settings - Fork 5
/
main.go
105 lines (84 loc) · 1.75 KB
/
main.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
103
104
105
package main
import (
"context"
"flag"
"os"
"os/signal"
"syscall"
"time"
"github.com/ecadlabs/rosdump/config"
"github.com/ecadlabs/rosdump/scraper"
log "github.com/sirupsen/logrus"
)
func runScraper(ctx context.Context, s *scraper.Scraper, timeout time.Duration) error {
log.Info("collecting data...")
if timeout != 0 {
ctx, _ = context.WithTimeout(ctx, timeout)
}
return s.Do(ctx)
}
func main() {
var (
configFile string
daemon bool
nowait bool
)
flag.StringVar(&configFile, "c", "", "Config")
flag.BoolVar(&daemon, "d", false, "Run in daemon mode")
flag.BoolVar(&nowait, "n", false, "Don't wait before first run")
flag.Parse()
if configFile == "" {
flag.Usage()
os.Exit(0)
}
cfg, err := config.Load(configFile)
if err != nil {
log.Fatal(err)
}
sc, err := scraper.New(cfg, log.StandardLogger())
if err != nil {
log.Fatal(err)
}
timeout, _ := time.ParseDuration(cfg.Timeout)
interval, _ := time.ParseDuration(cfg.Interval)
if daemon && interval == 0 {
log.Fatal("Interval must not be zero")
}
ctx, cancel := context.WithCancel(context.Background())
sem := make(chan struct{})
go func() {
defer close(sem)
if !daemon || nowait {
if err := runScraper(ctx, sc, timeout); err != nil {
if !daemon {
log.Fatal(err)
} else {
log.Error(err)
}
}
if !daemon {
os.Exit(0)
}
}
tick := time.Tick(interval)
for {
select {
case <-tick:
if err := runScraper(ctx, sc, timeout); err != nil {
log.Error(err)
}
case <-ctx.Done():
return
}
}
}()
signalChan := make(chan os.Signal, 1)
signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
select {
case s := <-signalChan:
log.Printf("captured %v\n", s)
cancel()
}
<-sem
log.Info("done")
}