forked from jrallison/go-workers
-
Notifications
You must be signed in to change notification settings - Fork 1
/
middleware_stats.go
37 lines (27 loc) · 692 Bytes
/
middleware_stats.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
package workers
import (
"time"
)
type MiddlewareStats struct{}
func (l *MiddlewareStats) Call(queue string, message *Msg, next func() bool) (acknowledge bool) {
defer func() {
if e := recover(); e != nil {
incrementStats("failed")
panic(e)
}
}()
acknowledge = next()
incrementStats("processed")
return
}
func incrementStats(metric string) {
conn := Config.Pool.Get()
defer conn.Close()
today := time.Now().UTC().Format("2006-01-02")
conn.Send("multi")
conn.Send("incr", Config.Namespace+"stat:"+metric)
conn.Send("incr", Config.Namespace+"stat:"+metric+":"+today)
if _, err := conn.Do("exec"); err != nil {
Logger.Println("couldn't save stats:", err)
}
}