mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-05 16:15:56 +00:00
45fd4fc992
This commit moves ID-related types and functions to the top level package and moves remaining statsd wrapper into a `statsd` package.
65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package statsd
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/cactus/go-statsd-client/statsd"
|
|
log "github.com/cihub/seelog"
|
|
)
|
|
|
|
// DefaultHost is the default host to send statsd data to.
|
|
const DefaultHost = "127.0.0.1:8125"
|
|
|
|
// Host is the host to send statsd data to.
|
|
var Host = DefaultHost
|
|
|
|
type args struct {
|
|
stat string
|
|
delta int64
|
|
rate float32
|
|
}
|
|
|
|
var timer chan args
|
|
var count chan string
|
|
var end chan bool
|
|
|
|
func Setup() {
|
|
timer = make(chan args, 32768)
|
|
count = make(chan string, 32768)
|
|
end = make(chan bool)
|
|
|
|
log.Warn("New Stats", Host)
|
|
stats, _ := statsd.New(Host, "")
|
|
|
|
go func() {
|
|
for {
|
|
select {
|
|
case ci := <-timer:
|
|
stats.Gauge(ci.stat, ci.delta, ci.rate)
|
|
stats.Timing(ci.stat, ci.delta, ci.rate)
|
|
case stat := <-count:
|
|
stats.Inc(stat, 1, 1.0)
|
|
case <-end:
|
|
log.Warn("DONE Stats")
|
|
return
|
|
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func SendTimer(stat string, delta int64) {
|
|
pstat := "pilosa." + stat
|
|
milli := time.Duration(delta) / time.Millisecond
|
|
timer <- args{pstat, int64(milli), 1.0}
|
|
}
|
|
|
|
func SendInc(stat string) {
|
|
pstat := "pilosa." + stat
|
|
count <- pstat
|
|
}
|
|
|
|
func ShutdownStats() {
|
|
log.Warn("Shutdown Stats")
|
|
end <- true
|
|
}
|