Merge pull request #401 from jzelinskie/style-nitpicks

Style nitpicks

bittorrent/bittorrent_test.go

@@ -19,22 +19,16 @@ var peerStringTestCases = []struct {
 }{
 	{
 		input: Peer{
-			ID: PeerIDFromBytes(b),
-			IP: IP{
-				IP:            net.IPv4(10, 11, 12, 1),
-				AddressFamily: IPv4,
-			},
+			ID:   PeerIDFromBytes(b),
+			IP:   IP{net.IPv4(10, 11, 12, 1), IPv4},
 			Port: 1234,
 		},
 		expected: fmt.Sprintf("%s@[10.11.12.1]:1234", expected),
 	},
 	{
 		input: Peer{
-			ID: PeerIDFromBytes(b),
-			IP: IP{
-				IP:            net.ParseIP("2001:db8::ff00:42:8329"),
-				AddressFamily: IPv6,
-			},
+			ID:   PeerIDFromBytes(b),
+			IP:   IP{net.ParseIP("2001:db8::ff00:42:8329"), IPv6},
 			Port: 1234,
 		},
 		expected: fmt.Sprintf("%s@[2001:db8::ff00:42:8329]:1234", expected),

frontend/http/frontend.go

@@ -10,52 +10,12 @@ import (
 	"time"
 
 	"github.com/julienschmidt/httprouter"
-	"github.com/prometheus/client_golang/prometheus"
 
 	"github.com/chihaya/chihaya/bittorrent"
 	"github.com/chihaya/chihaya/frontend"
 	"github.com/chihaya/chihaya/pkg/log"
 )
 
-func init() {
-	prometheus.MustRegister(promResponseDurationMilliseconds)
-}
-
-var promResponseDurationMilliseconds = prometheus.NewHistogramVec(
-	prometheus.HistogramOpts{
-		Name:    "chihaya_http_response_duration_milliseconds",
-		Help:    "The duration of time it takes to receive and write a response to an API request",
-		Buckets: prometheus.ExponentialBuckets(9.375, 2, 10),
-	},
-	[]string{"action", "address_family", "error"},
-)
-
-// recordResponseDuration records the duration of time to respond to a Request
-// in milliseconds .
-func recordResponseDuration(action string, af *bittorrent.AddressFamily, err error, duration time.Duration) {
-	var errString string
-	if err != nil {
-		if _, ok := err.(bittorrent.ClientError); ok {
-			errString = err.Error()
-		} else {
-			errString = "internal error"
-		}
-	}
-
-	var afString string
-	if af == nil {
-		afString = "Unknown"
-	} else if *af == bittorrent.IPv4 {
-		afString = "IPv4"
-	} else if *af == bittorrent.IPv6 {
-		afString = "IPv6"
-	}
-
-	promResponseDurationMilliseconds.
-		WithLabelValues(action, afString, errString).
-		Observe(float64(duration.Nanoseconds()) / float64(time.Millisecond))
-}
-
 // Config represents all of the configurable options for an HTTP BitTorrent
 // Frontend.
 type Config struct {

frontend/http/prometheus.go

@@ -0,0 +1,48 @@
+package http
+
+import (
+	"time"
+
+	"github.com/prometheus/client_golang/prometheus"
+
+	"github.com/chihaya/chihaya/bittorrent"
+)
+
+func init() {
+	prometheus.MustRegister(promResponseDurationMilliseconds)
+}
+
+var promResponseDurationMilliseconds = prometheus.NewHistogramVec(
+	prometheus.HistogramOpts{
+		Name:    "chihaya_http_response_duration_milliseconds",
+		Help:    "The duration of time it takes to receive and write a response to an API request",
+		Buckets: prometheus.ExponentialBuckets(9.375, 2, 10),
+	},
+	[]string{"action", "address_family", "error"},
+)
+
+// recordResponseDuration records the duration of time to respond to a Request
+// in milliseconds.
+func recordResponseDuration(action string, af *bittorrent.AddressFamily, err error, duration time.Duration) {
+	var errString string
+	if err != nil {
+		if _, ok := err.(bittorrent.ClientError); ok {
+			errString = err.Error()
+		} else {
+			errString = "internal error"
+		}
+	}
+
+	var afString string
+	if af == nil {
+		afString = "Unknown"
+	} else if *af == bittorrent.IPv4 {
+		afString = "IPv4"
+	} else if *af == bittorrent.IPv6 {
+		afString = "IPv6"
+	}
+
+	promResponseDurationMilliseconds.
+		WithLabelValues(action, afString, errString).
+		Observe(float64(duration.Nanoseconds()) / float64(time.Millisecond))
+}

frontend/udp/frontend.go

@@ -12,8 +12,6 @@ import (
 	"sync"
 	"time"
 
-	"github.com/prometheus/client_golang/prometheus"
-
 	"github.com/chihaya/chihaya/bittorrent"
 	"github.com/chihaya/chihaya/frontend"
 	"github.com/chihaya/chihaya/frontend/udp/bytepool"
@@ -24,45 +22,6 @@ import (
 
 var allowedGeneratedPrivateKeyRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890")
 
-func init() {
-	prometheus.MustRegister(promResponseDurationMilliseconds)
-}
-
-var promResponseDurationMilliseconds = prometheus.NewHistogramVec(
-	prometheus.HistogramOpts{
-		Name:    "chihaya_udp_response_duration_milliseconds",
-		Help:    "The duration of time it takes to receive and write a response to an API request",
-		Buckets: prometheus.ExponentialBuckets(9.375, 2, 10),
-	},
-	[]string{"action", "address_family", "error"},
-)
-
-// recordResponseDuration records the duration of time to respond to a UDP
-// Request in milliseconds .
-func recordResponseDuration(action string, af *bittorrent.AddressFamily, err error, duration time.Duration) {
-	var errString string
-	if err != nil {
-		if _, ok := err.(bittorrent.ClientError); ok {
-			errString = err.Error()
-		} else {
-			errString = "internal error"
-		}
-	}
-
-	var afString string
-	if af == nil {
-		afString = "Unknown"
-	} else if *af == bittorrent.IPv4 {
-		afString = "IPv4"
-	} else if *af == bittorrent.IPv6 {
-		afString = "IPv6"
-	}
-
-	promResponseDurationMilliseconds.
-		WithLabelValues(action, afString, errString).
-		Observe(float64(duration.Nanoseconds()) / float64(time.Millisecond))
-}
-
 // Config represents all of the configurable options for a UDP BitTorrent
 // Tracker.
 type Config struct {

frontend/udp/prometheus.go

@@ -0,0 +1,48 @@
+package udp
+
+import (
+	"time"
+
+	"github.com/prometheus/client_golang/prometheus"
+
+	"github.com/chihaya/chihaya/bittorrent"
+)
+
+func init() {
+	prometheus.MustRegister(promResponseDurationMilliseconds)
+}
+
+var promResponseDurationMilliseconds = prometheus.NewHistogramVec(
+	prometheus.HistogramOpts{
+		Name:    "chihaya_udp_response_duration_milliseconds",
+		Help:    "The duration of time it takes to receive and write a response to an API request",
+		Buckets: prometheus.ExponentialBuckets(9.375, 2, 10),
+	},
+	[]string{"action", "address_family", "error"},
+)
+
+// recordResponseDuration records the duration of time to respond to a UDP
+// Request in milliseconds.
+func recordResponseDuration(action string, af *bittorrent.AddressFamily, err error, duration time.Duration) {
+	var errString string
+	if err != nil {
+		if _, ok := err.(bittorrent.ClientError); ok {
+			errString = err.Error()
+		} else {
+			errString = "internal error"
+		}
+	}
+
+	var afString string
+	if af == nil {
+		afString = "Unknown"
+	} else if *af == bittorrent.IPv4 {
+		afString = "IPv4"
+	} else if *af == bittorrent.IPv6 {
+		afString = "IPv6"
+	}
+
+	promResponseDurationMilliseconds.
+		WithLabelValues(action, afString, errString).
+		Observe(float64(duration.Nanoseconds()) / float64(time.Millisecond))
+}

storage/memory/peer_store_test.go

@@ -9,7 +9,12 @@ import (
 )
 
 func createNew() s.PeerStore {
-	ps, err := New(Config{ShardCount: 1024, GarbageCollectionInterval: 10 * time.Minute, PrometheusReportingInterval: 10 * time.Minute, PeerLifetime: 30 * time.Minute})
+	ps, err := New(Config{
+		ShardCount:                  1024,
+		GarbageCollectionInterval:   10 * time.Minute,
+		PrometheusReportingInterval: 10 * time.Minute,
+		PeerLifetime:                30 * time.Minute,
+	})
 	if err != nil {
 		panic(err)
 	}

storage/memorybysubnet/peer_store.go

@@ -520,7 +520,7 @@ func (ps *peerStore) AnnouncePeers(ih bittorrent.InfoHash, seeder bool, numWant
 		// Append the rest of the leechers.
 		if numWant > 0 {
 			for subnet := range shard.swarms[ih].leechers {
-				// Already appended from this subnet explictly first.
+				// Already appended from this subnet explicitly first.
 				if subnet == preferredSubnet {
 					continue
 				}
@@ -568,7 +568,7 @@ func (ps *peerStore) AnnouncePeers(ih bittorrent.InfoHash, seeder bool, numWant
 		// Append as the rest of the seeders.
 		if numWant > 0 {
 			for subnet := range shard.swarms[ih].seeders {
-				// Already appended from this subnet explictly first.
+				// Already appended from this subnet explicitly first.
 				if subnet == preferredSubnet {
 					continue
 				}
@@ -587,7 +587,7 @@ func (ps *peerStore) AnnouncePeers(ih bittorrent.InfoHash, seeder bool, numWant
 		// Append the rest of the leechers.
 		if numWant > 0 {
 			for subnet := range shard.swarms[ih].leechers {
-				// Already appended from this subnet explictly first.
+				// Already appended from this subnet explicitly first.
 				if subnet == preferredSubnet {
 					continue
 				}