mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-09-05 16:15:56 +00:00
2f1e3b6078
This commit refactors the config into a `main.Config` object instead of a global singleton. The `core.Service` is also refactored into the `main` package and individual pieces of the service are wired together by the `pilosa` binary. These two changes are required to begin to decouple packages from one another and allow them to be individually unit tested. Previously most top-level objects in the system could access any other top-level object through the `core.Service` which effectively made `Service` a singleton in the system. Each top-level object now has inline interfaces for their dependencies so that can be set at runtime by the `main` package or can be mocked by a test package.
67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package util
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/cactus/go-statsd-client/statsd"
|
|
log "github.com/cihub/seelog"
|
|
)
|
|
|
|
// DefaultStatsdHost is the default host to send statsd data to.
|
|
const DefaultStatsdHost = "127.0.0.1:8125"
|
|
|
|
// StatsdHost is the host to send statsd data to.
|
|
var StatsdHost = DefaultStatsdHost
|
|
|
|
type args struct {
|
|
stat string
|
|
delta int64
|
|
rate float32
|
|
}
|
|
|
|
var (
|
|
timer chan args
|
|
count chan string
|
|
end chan bool
|
|
)
|
|
|
|
func SetupStatsd() {
|
|
timer = make(chan args, 32768)
|
|
count = make(chan string, 32768)
|
|
end = make(chan bool)
|
|
|
|
log.Warn("New Stats", StatsdHost)
|
|
stats, _ := statsd.New(StatsdHost, "")
|
|
|
|
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
|
|
}
|