-
Notifications
You must be signed in to change notification settings - Fork 2
/
main.go
110 lines (91 loc) · 2.17 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
106
107
108
109
110
package main
import (
"fmt"
"log"
"net/http"
_ "net/http/pprof"
"os"
"strconv"
"time"
)
const (
defaultCollectMetrics = true
defaultCollectEvents = true
)
var (
collectMetrics bool
collectEvents bool
)
func handler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "All good yo")
}
func main() {
log.Println("Starting")
collectMetricsStr := os.Getenv("COLLECT_METRICS")
collectEventsStr := os.Getenv("COLLECT_EVENTS")
if b, err := strconv.ParseBool(collectMetricsStr); err == nil {
collectMetrics = b
} else {
collectMetrics = defaultCollectMetrics
}
if b, err := strconv.ParseBool(collectEventsStr); err == nil {
collectEvents = b
} else {
collectEvents = defaultCollectEvents
}
events_collector_chan := make(chan *Events)
metrics_collector_chan := make(chan *Metrics)
events_reciever_chans := make(map[EventReciever]chan *Events)
metrics_reciever_chans := make(map[MetricReciever]chan *Metrics)
events_recievers := LoadEventRecievers()
metrics_recievers := LoadMetricsRecievers()
for _, v := range events_recievers {
events_reciever_chans[v] = make(chan *Events)
}
for _, v := range metrics_recievers {
metrics_reciever_chans[v] = make(chan *Metrics)
}
if collectEvents {
log.Println("Starting event collection")
event_collectors := LoadEventCollectors()
for _, v := range event_collectors {
if v != nil {
go EventProducer(v, events_collector_chan)
}
}
for v, c := range events_reciever_chans {
go v.Run(c)
}
}
if collectMetrics {
log.Println("Starting metric collection")
metric_collectors := LoadMetricCollectors()
for _, v := range metric_collectors {
if v != nil {
go MetricProducer(v, metrics_collector_chan)
}
}
for v, c := range metrics_reciever_chans {
go v.Run(c)
}
}
go func() {
http.HandleFunc("/", handler)
log.Println(http.ListenAndServe(":8080", nil))
}()
for {
select {
case e := <-events_collector_chan:
for _, re := range events_reciever_chans {
re <- e
}
case m := <-metrics_collector_chan:
for _, rm := range metrics_reciever_chans {
rm <- m
}
default:
// Do this sleep to cut back on cpu usage
time.Sleep(1 * time.Millisecond)
}
}
}