From bd1fa3eb24bd6866ae1ef8746e9bbad4df78e2dc Mon Sep 17 00:00:00 2001
From: Jimmy Zelinskie <jimmyzelinskie@gmail.com>
Date: Wed, 23 Jul 2014 03:59:23 -0400
Subject: [PATCH] cli flag for default stats chan size

---
 stats/stats.go | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/stats/stats.go b/stats/stats.go
index d9f7577..2ea809a 100644
--- a/stats/stats.go
+++ b/stats/stats.go
@@ -6,7 +6,10 @@
 // BitTorrent tracker.
 package stats
 
-import "time"
+import (
+	"flag"
+	"time"
+)
 
 const (
 	Announce = iota
@@ -34,11 +37,15 @@ const (
 )
 
 // DefaultStats is a default instance of stats tracking that uses an unbuffered
-// channel for broadcasting events.
-var DefaultStats *Stats
+// channel for broadcasting events unless specified otherwise via a command
+// line flag.
+var (
+	DefaultStats     *Stats
+	DefaultChanSizes int
+)
 
 func init() {
-	DefaultStats = New(0)
+	flag.IntVar(&DefaultChanSizes, "stats_chan_sizes", 0, "specifies the size of chans used to collect stats")
 }
 
 type PeerStats struct {
@@ -238,15 +245,27 @@ func (s *Stats) handlePeerEvent(ps *PeerStats, event int) {
 
 // RecordEvent broadcasts an event to the default stats queue.
 func RecordEvent(event int) {
+	if DefaultStats == nil {
+		DefaultStats = New(DefaultChanSizes)
+	}
+
 	DefaultStats.RecordEvent(event)
 }
 
 // RecordPeerEvent broadcasts a peer event to the default stats queue.
 func RecordPeerEvent(event int, ipv6 bool) {
+	if DefaultStats == nil {
+		DefaultStats = New(DefaultChanSizes)
+	}
+
 	DefaultStats.RecordPeerEvent(event, ipv6)
 }
 
 // RecordTiming broadcasts a timing event to the default stats queue.
 func RecordTiming(event int, duration time.Duration) {
+	if DefaultStats == nil {
+		DefaultStats = New(DefaultChanSizes)
+	}
+
 	DefaultStats.RecordTiming(event, duration)
 }