client_bench_test.go

package statsd

import (
	"testing"
	"time"
)

const (
	host        = "localhost"
	port        = 8125
	prefix      = "prefix."
	prefixNoDot = "prefix"
	counterKey  = "foo.bar.counter"
	gaugeKey    = "foo.bar.gauge"
	gaugeValue  = 42
	timingKey   = "foo.bar.timing"
	timingValue = 153 * time.Millisecond
	flushPeriod = 10 * time.Millisecond
)

func BenchmarkBuffered(b *testing.B) {
	client := NewBufferedClient(host, port)
	client.Open()
	client.SetPrefix(prefix)
	client.SetFlushInterval(flushPeriod)

	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		client.Count(counterKey, 1, 1)
		client.Gauge(gaugeKey, gaugeValue)
		client.Timing(timingKey, int64(timingValue), 1)
	}

	client.Close()
}

func BenchmarkNotBuffered(b *testing.B) {
	client := NewClient(host, port)
	client.SetPrefix(prefix)
	client.Open()

	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		client.Count(counterKey, 1, 1)
		client.Gauge(gaugeKey, gaugeValue)
		client.Timing(timingKey, int64(timingValue), 1)
	}

	client.Close()
}