use cactus as statsd client

Author: sokil

go.mod

@@ -4,7 +4,9 @@ go 1.12
 
 require (
 	github.com/GoMetric/go-statsd-client v1.0.2
+	github.com/cactus/go-statsd-client/v5 v5.0.0
 	github.com/dgrijalva/jwt-go v3.2.0+incompatible
+	github.com/golang/protobuf v1.5.2
 	github.com/gorilla/mux v1.7.3
 	github.com/julienschmidt/httprouter v1.3.0
 	github.com/stretchr/testify v1.4.0

go.sum

@@ -3,12 +3,21 @@ github.com/GoMetric/go-statsd-client v0.0.0-20191006202127-dc032ac6197b/go.mod h
 github.com/GoMetric/go-statsd-client v1.0.0/go.mod h1:D1uEs99MnXBDuAkJMdegj5SxBsBzOctG06Wfxyjr8jE=
 github.com/GoMetric/go-statsd-client v1.0.2 h1:HgAeuEAbzVlQGLNqjuGzczqbYwpStPcTjiO0v71ModE=
 github.com/GoMetric/go-statsd-client v1.0.2/go.mod h1:B44qfLbrIFTm9+Xhqhep07th+soi+/t8g9HKwUvDlAk=
+github.com/cactus/go-statsd-client v1.0.1 h1:R58kYEe8CS4Dg7zNgGi8J0nFhYIGaBvIC7SHJ1l08is=
+github.com/cactus/go-statsd-client v3.2.1+incompatible h1:hSd8ARQzCrnKQUWKdOBbwhU1nGR/al1oiT1AeBzoaTY=
+github.com/cactus/go-statsd-client/v5 v5.0.0 h1:KqvIQtc9qt34uq+nu4nd1PwingWfBt/IISgtUQ2nSJk=
+github.com/cactus/go-statsd-client/v5 v5.0.0/go.mod h1:COEvJ1E+/E2L4q6QE5CkjWPi4eeDw9maJBMIuMPBZbY=
 github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM=
 github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
+github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
+github.com/golang/protobuf v1.5.2 h1:ROPKBNFfQgOUMifHyP+KYbvpjbdoFNs+aK7DXlji0Tw=
+github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/gorilla/mux v1.7.3 h1:gnP5JzjVOuiZD07fKKToCAOjS0yOpj/qPETTXCCS6hw=
 github.com/gorilla/mux v1.7.3/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
+github.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=
 github.com/julienschmidt/httprouter v1.3.0 h1:U0609e9tgbseu3rBINet9P48AI/D3oJs4dN7jwJOQ1U=
 github.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
@@ -17,6 +26,10 @@ github.com/stretchr/objx v0.1.0 h1:4G4v2dO3VZwixGIRoQ5Lfboy6nUhCyYzaqnIAPPhYs4=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
+google.golang.org/protobuf v1.26.0 h1:bxAC2xTBsZGibn2RTntX0oH50xLsqy1OxA9tTL3p/lk=
+google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
 gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

proxy/server.go

@@ -55,7 +55,17 @@ func NewServer(
 	logger := log.New(logOutput, "", log.LstdFlags)
 
 	// create StatsD Client
-	statsdClient := statsdclient.NewGoMetricClient(statsdHost, statsdPort)
+	statsdClientName := "Cactus"
+
+	var statsdClient statsdclient.StatsdClientInterface
+	switch statsdClientName {
+	case "GoMetric":
+		statsdClient = statsdclient.NewGoMetricClient(statsdHost, statsdPort)
+	case "Cactus":
+		statsdClient = statsdclient.NewCactusClient(statsdHost, statsdPort)
+	default:
+		panic("Passed statsd client not supported")
+	}
 
 	// build route handler
 	routeHandler := routehandler.NewRouteHandler(

proxy/statsdclient/cactus.go

@@ -0,0 +1,56 @@
+package statsdclient
+
+import (
+	"strconv"
+	"time"
+
+	cactus "github.com/cactus/go-statsd-client/v5/statsd"
+)
+
+type CactusStatsdClientAdapter struct {
+	cactusClient cactus.Statter
+}
+
+func (a *CactusStatsdClientAdapter) Open() {
+
+}
+
+func (a *CactusStatsdClientAdapter) Close() {
+	a.cactusClient.Close()
+}
+
+func (a *CactusStatsdClientAdapter) Count(key string, value int, sampleRate float32) {
+	a.cactusClient.Inc(key, int64(value), sampleRate)
+}
+
+func (a *CactusStatsdClientAdapter) Timing(key string, time int64, sampleRate float32) {
+	a.cactusClient.Timing(key, time, sampleRate)
+}
+
+func (a *CactusStatsdClientAdapter) Gauge(key string, value int) {
+	a.cactusClient.Gauge(key, int64(value), 1)
+}
+
+func (a *CactusStatsdClientAdapter) GaugeShift(key string, value int) {
+	a.cactusClient.GaugeDelta(key, int64(value), 1)
+}
+
+func (a *CactusStatsdClientAdapter) Set(key string, value int) {
+	a.cactusClient.SetInt(key, int64(value), 1)
+}
+
+func NewCactusClient(
+	statsdHost string,
+	statsdPort int,
+) StatsdClientInterface {
+	cactusClient, _ := cactus.NewBufferedClient(
+		statsdHost+":"+strconv.Itoa(statsdPort),
+		"",
+		100*time.Microsecond,
+		1432,
+	)
+
+	return &CactusStatsdClientAdapter{
+		cactusClient: cactusClient,
+	}
+}