// Metrics output to StatHat.
package stathat

import (
	"github.com/rcrowley/go-metrics"
	"github.com/stathat/go"
	"log"
	"time"
)

func Stathat(r metrics.Registry, d time.Duration, userkey string) {
	for {
		if err := sh(r, userkey); nil != err {
			log.Println(err)
		}
		time.Sleep(d)
	}
}

func sh(r metrics.Registry, userkey string) error {
	r.Each(func(name string, i interface{}) {
		switch m := i.(type) {
		case metrics.Counter:
			stathat.PostEZCount(name, userkey, int(m.Count()))
		case metrics.Gauge:
			stathat.PostEZValue(name, userkey, float64(m.Value()))
		case metrics.Histogram:
			s := m.Sample()
			ps := s.Percentiles([]float64{0.5, 0.75, 0.95, 0.99, 0.999})
			stathat.PostEZCount(name+".count", userkey, int(s.Count()))
			stathat.PostEZValue(name+".min", userkey, float64(s.Min()))
			stathat.PostEZValue(name+".max", userkey, float64(s.Max()))
			stathat.PostEZValue(name+".mean", userkey, float64(s.Mean()))
			stathat.PostEZValue(name+".std-dev", userkey, float64(s.StdDev()))
			stathat.PostEZValue(name+".50-percentile", userkey, float64(ps[0]))
			stathat.PostEZValue(name+".75-percentile", userkey, float64(ps[1]))
			stathat.PostEZValue(name+".95-percentile", userkey, float64(ps[2]))
			stathat.PostEZValue(name+".99-percentile", userkey, float64(ps[3]))
			stathat.PostEZValue(name+".999-percentile", userkey, float64(ps[4]))
		case metrics.Meter:
			stathat.PostEZCount(name+".count", userkey, int(m.Count()))
			stathat.PostEZValue(name+".one-minute", userkey, float64(m.Rate1()))
			stathat.PostEZValue(name+".five-minute", userkey, float64(m.Rate5()))
			stathat.PostEZValue(name+".fifteen-minute", userkey, float64(m.Rate15()))
			stathat.PostEZValue(name+".mean", userkey, float64(m.RateMean()))
		case metrics.Timer:
			ps := m.Percentiles([]float64{0.5, 0.75, 0.95, 0.99, 0.999})
			stathat.PostEZCount(name+".count", userkey, int(m.Count()))
			stathat.PostEZValue(name+".min", userkey, float64(m.Min()))
			stathat.PostEZValue(name+".max", userkey, float64(m.Max()))
			stathat.PostEZValue(name+".mean", userkey, float64(m.Mean()))
			stathat.PostEZValue(name+".std-dev", userkey, float64(m.StdDev()))
			stathat.PostEZValue(name+".50-percentile", userkey, float64(ps[0]))
			stathat.PostEZValue(name+".75-percentile", userkey, float64(ps[1]))
			stathat.PostEZValue(name+".95-percentile", userkey, float64(ps[2]))
			stathat.PostEZValue(name+".99-percentile", userkey, float64(ps[3]))
			stathat.PostEZValue(name+".999-percentile", userkey, float64(ps[4]))
			stathat.PostEZValue(name+".one-minute", userkey, float64(m.Rate1()))
			stathat.PostEZValue(name+".five-minute", userkey, float64(m.Rate5()))
			stathat.PostEZValue(name+".fifteen-minute", userkey, float64(m.Rate15()))
			stathat.PostEZValue(name+".mean-rate", userkey, float64(m.RateMean()))
		}
	})
	return nil
}