middleware_stats.go

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")

	err := conn.Send("multi")
	if err != nil {
		Logger.Println("couldn't send multi:", err)
	}
	err = conn.Send("incr", Config.Namespace+"stat:"+metric)
	if err != nil {
		Logger.Println("could not incr stat:", err)
	}
	err = conn.Send("incr", Config.Namespace+"stat:"+metric+":"+today)
	if err != nil {
		Logger.Println("could not incr stat for today:", err)
	}

	if _, err := conn.Do("exec"); err != nil {
		Logger.Println("couldn't save stats:", err)
	}
}