func send_gauge(client statsd.Statter, v []string, name string, position int64) {
stat := fmt.Sprint(v[0], ".", name)
value, _ := strconv.ParseInt(v[position], 10, 64)
fmt.Println(fmt.Sprint(stat, ":", value, "|g"))
err := client.Gauge(stat, value, 1.0)
if err != nil {
log.Printf("Error sending metric: %+v", err)
}
}
func pushMem(client statsd.Statter, mem Mem) error {
var err error
err = client.Gauge("mem.resident", mem.Resident, 1.0)
if err != nil {
return err
}
err = client.Gauge("mem.virtual", mem.Virtual, 1.0)
if err != nil {
return err
}
err = client.Gauge("mem.mapped", mem.Mapped, 1.0)
if err != nil {
return err
}
err = client.Gauge("mem.mapped_with_journal", mem.MappedWithJournal, 1.0)
if err != nil {
return err
}
return nil
}
func pushExtraInfo(client statsd.Statter, info ExtraInfo) error {
var err error
err = client.Gauge("extra.page_faults", info.PageFaults, 1.0)
if err != nil {
return err
}
err = client.Gauge("extra.heap_usage", info.HeapUsageInBytes, 1.0)
if err != nil {
return err
}
return nil
}
func pushConnections(client statsd.Statter, connections Connections) error {
var err error
// Connections
err = client.Gauge("connections.current", int64(connections.Current), 1.0)
if err != nil {
return err
}
err = client.Gauge("connections.available", int64(connections.Available), 1.0)
if err != nil {
return err
}
err = client.Gauge("connections.created", int64(connections.TotalCreated), 1.0)
if err != nil {
return err
}
return nil
}
func sendGauges(client statsd.Statter, gauges map[string]int64) {
for gauge, value := range gauges {
client.Gauge(gauge, value, 1.0)
}
}
func main() {
// command line flags
var opts struct {
HostPort string `long:"host" default:"127.0.0.1:8125" description:"host:port of statsd server"`
Prefix string `long:"prefix" default:"test-client" description:"Statsd prefix"`
StatType string `long:"type" default:"count" description:"stat type to send. Can be timing, count, guage"`
StatValue int64 `long:"value" default:"1" description:"Value to send"`
Name string `short:"n" long:"name" default:"counter" description:"stat name"`
Rate float32 `short:"r" long:"rate" default:"1.0" description:"sample rate"`
Volume int `short:"c" long:"count" default:"1000" description:"Number of stats to send. Volume."`
Noop bool `long:"noop" default:"false" description:"Use noop client"`
Duration time.Duration `short:"d" long:"duration" default:"10s" description:"How long to spread the volume across. Each second of duration volume/seconds events will be sent."`
}
// parse said flags
_, err := flags.Parse(&opts)
if err != nil {
if e, ok := err.(*flags.Error); ok {
if e.Type == flags.ErrHelp {
os.Exit(0)
}
}
fmt.Printf("Error: %+v\n", err)
os.Exit(1)
}
var client statsd.Statter
if !opts.Noop {
client, err = statsd.New(opts.HostPort, opts.Prefix)
if err != nil {
log.Fatal(err)
}
defer client.Close()
} else {
client, err = statsd.NewNoop(opts.HostPort, opts.Prefix)
}
var stat func(stat string, value int64, rate float32) error
switch opts.StatType {
case "count":
stat = func(stat string, value int64, rate float32) error {
return client.Inc(stat, value, rate)
}
case "gauge":
stat = func(stat string, value int64, rate float32) error {
return client.Gauge(stat, value, rate)
}
case "timing":
stat = func(stat string, value int64, rate float32) error {
return client.Timing(stat, value, rate)
}
default:
log.Fatal("Unsupported state type")
}
pertick := opts.Volume / int(opts.Duration.Seconds()) / 10
// add some extra tiem, because the first tick takes a while
ender := time.After(opts.Duration + 100*time.Millisecond)
c := time.Tick(time.Second / 10)
count := 0
for {
select {
case <-c:
for x := 0; x < pertick; x++ {
err := stat(opts.Name, opts.StatValue, opts.Rate)
if err != nil {
log.Printf("Got Error: %+v", err)
break
}
count += 1
}
case <-ender:
log.Printf("%d events called", count)
os.Exit(0)
return
}
}
}
func pushGlobalLocks(client statsd.Statter, glob GlobalLock) error {
var err error
err = client.Gauge("global_lock.total_time", glob.TotalTime, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.lock_time", glob.LockTime, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.active_readers", glob.ActiveClients.Readers, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.active_writers", glob.ActiveClients.Writers, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.active_total", glob.ActiveClients.Total, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.queued_readers", glob.CurrentQueue.Readers, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.queued_writers", glob.CurrentQueue.Writers, 1.0)
if err != nil {
return err
}
err = client.Gauge("global_lock.queued_total", glob.CurrentQueue.Total, 1.0)
if err != nil {
return err
}
return nil
}
func pushOpcounters(client statsd.Statter, opscounters Opcounters) error {
var err error
// Ops Counters (non-RS)
err = client.Gauge("ops.inserts", opscounters.Insert, 1.0)
if err != nil {
return err
}
err = client.Gauge("ops.queries", opscounters.Query, 1.0)
if err != nil {
return err
}
err = client.Gauge("ops.updates", opscounters.Update, 1.0)
if err != nil {
return err
}
err = client.Gauge("ops.deletes", opscounters.Delete, 1.0)
if err != nil {
return err
}
err = client.Gauge("ops.getmores", opscounters.GetMore, 1.0)
if err != nil {
return err
}
err = client.Gauge("ops.commands", opscounters.Command, 1.0)
if err != nil {
return err
}
return nil
}
// ProfileCmd runs forever, sending Go runtime statistics to StatsD.
func ProfileCmd(stats statsd.Statter) {
c := time.Tick(1 * time.Second)
for range c {
var memoryStats runtime.MemStats
runtime.ReadMemStats(&memoryStats)
stats.Gauge("Gostats.Goroutines", int64(runtime.NumGoroutine()), 1.0)
stats.Gauge("Gostats.Heap.Alloc", int64(memoryStats.HeapAlloc), 1.0)
stats.Gauge("Gostats.Heap.Objects", int64(memoryStats.HeapObjects), 1.0)
stats.Gauge("Gostats.Heap.Idle", int64(memoryStats.HeapIdle), 1.0)
stats.Gauge("Gostats.Heap.InUse", int64(memoryStats.HeapInuse), 1.0)
stats.Gauge("Gostats.Heap.Released", int64(memoryStats.HeapReleased), 1.0)
// Calculate average and last and convert from nanoseconds to milliseconds
gcPauseAvg := (int64(memoryStats.PauseTotalNs) / int64(len(memoryStats.PauseNs))) / 1000000
lastGC := int64(memoryStats.PauseNs[(memoryStats.NumGC+255)%256]) / 1000000
stats.Timing("Gostats.Gc.PauseAvg", gcPauseAvg, 1.0)
stats.Gauge("Gostats.Gc.LastPauseTook", lastGC, 1.0)
stats.Gauge("Gostats.Gc.NextAt", int64(memoryStats.NextGC), 1.0)
}
}