add instance type tag on autothrottle configuration

Author: zhaolong

cmd/autothrottle/api.go

@@ -7,7 +7,7 @@ import (
 	"net/http"
 	"strconv"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkazk"
+	"github.com/DataDog/kafka-kit/kafkazk"
 )
 
 // APIConfig holds configuration

cmd/autothrottle/events.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 	"log"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics"
 )
 
 // Events configs.

cmd/autothrottle/helpers.go

@@ -4,7 +4,7 @@ import (
 	"bytes"
 	"sort"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics"
 )
 
 // bmapBundle holds several maps

cmd/autothrottle/limits.go

@@ -4,7 +4,7 @@ import (
 	"errors"
 	"math"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics"
 )
 
 // Limits is a map of instance-type

cmd/autothrottle/limits_test.go

@@ -3,7 +3,7 @@ package main
 import (
 	"testing"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics"
 )
 
 func TestNewLimits(t *testing.T) {

cmd/autothrottle/main.go

@@ -10,9 +10,9 @@ import (
 	"strings"
 	"time"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics/datadog"
-	"github.com/mrmuggymuggy/kafka-kit/kafkazk"
+	"github.com/DataDog/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics/datadog"
+	"github.com/DataDog/kafka-kit/kafkazk"
 
 	"github.com/jamiealquiza/envy"
 )

cmd/autothrottle/throttles.go

@@ -9,8 +9,8 @@ import (
 	"strconv"
 	"time"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
-	"github.com/mrmuggymuggy/kafka-kit/kafkazk"
+	"github.com/DataDog/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkazk"
 )
 
 // ReplicationThrottleMeta holds all types
@@ -127,6 +127,7 @@ func updateReplicationThrottle(params *ReplicationThrottleMeta) error {
 				inFailureMode = true
 			}
 		}
+
 		// If we cannot proceed normally due to missing/partial
 		// metrics data, check what failure iteration we're in.
 		// If we're above the threshold, revert to the minimum
@@ -289,6 +290,7 @@ func repCapacityByMetrics(rtm *ReplicationThrottleMeta, bmb bmapBundle, bm kafka
 	participatingBrokers := &ReassigningBrokers{}
 
 	var event string
+
 	// Source brokers.
 	for b := range bmb.src {
 		if broker, exists := bm[b]; exists {

cmd/autothrottle/throttles_test.go

@@ -5,8 +5,8 @@ import (
 	"sort"
 	"testing"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
-	"github.com/mrmuggymuggy/kafka-kit/kafkazk"
+	"github.com/DataDog/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkazk"
 )
 
 func TestHighestSrcNetTX(t *testing.T) {

kafkametrics/datadog/datadog.go

@@ -7,7 +7,7 @@ import (
 	"regexp"
 	"time"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics"
 
 	dd "github.com/zorkian/go-datadog-api"
 )
@@ -40,7 +40,7 @@ type ddHandler struct {
 	c               *dd.Client
 	netTXQuery      string
 	brokerIDTag     string
-	InstanceTypeTag string
+	instanceTypeTag string
 	metricsWindow   int
 	tagCache        map[string][]string
 	keysRegex       *regexp.Regexp
@@ -64,7 +64,7 @@ func NewHandler(c *Config) (kafkametrics.Handler, error) {
 		netTXQuery:      createNetTXQuery(c),
 		metricsWindow:   c.MetricsWindow,
 		brokerIDTag:     c.BrokerIDTag,
-		InstanceTypeTag: c.InstanceTypeTag,
+		instanceTypeTag: c.InstanceTypeTag,
 		tagCache:        make(map[string][]string),
 		keysRegex:       keysRegex,
 		redactionSub:    []byte("xxx"),
@@ -137,6 +137,7 @@ func (h *ddHandler) GetMetrics() (kafkametrics.BrokerMetrics, []error) {
 	if errs != nil {
 		errors = append(errors, errs...)
 	}
+
 	// The []*kafkametrics.Broker only contains hostnames
 	// and the network tx metric. Fetch the rest
 	// of the required metadata and construct

kafkametrics/datadog/datadog_test.go

@@ -6,7 +6,7 @@ import (
 	"fmt"
 	"testing"
 
-	"github.com/mrmuggymuggy/kafka-kit/kafkametrics"
+	"github.com/DataDog/kafka-kit/kafkametrics"
 
 	dd "github.com/zorkian/go-datadog-api"
 )