stop arbiter if all meters are stopped

Author: cenkalti

meter.go

@@ -38,13 +38,7 @@ func NewMeter() Meter {
 		return NilMeter{}
 	}
 	m := newStandardMeter()
-	arbiter.Lock()
-	defer arbiter.Unlock()
-	arbiter.meters[m] = struct{}{}
-	if !arbiter.started {
-		arbiter.started = true
-		go arbiter.tick()
-	}
+	m.startArbiter()
 	return m
 }
 
@@ -145,9 +139,7 @@ func newStandardMeter() *StandardMeter {
 // Stop stops the meter, Mark() will be a no-op if you use it after being stopped.
 func (m *StandardMeter) Stop() {
 	if atomic.CompareAndSwapUint32(&m.stopped, 0, 1) {
-		arbiter.Lock()
-		delete(arbiter.meters, m)
-		arbiter.Unlock()
+		m.stopArbiter()
 	}
 }
 
@@ -156,7 +148,7 @@ func (m *StandardMeter) Count() int64 {
 	return atomic.LoadInt64(&m.snapshot.count)
 }
 
-// Mark records the occurance of n events.
+// Mark records the occurrance of n events.
 func (m *StandardMeter) Mark(n int64) {
 	if atomic.LoadUint32(&m.stopped) == 1 {
 		return
@@ -221,23 +213,50 @@ func (m *StandardMeter) tick() {
 	m.updateSnapshot()
 }
 
+func (m *StandardMeter) startArbiter() {
+	arbiter.Lock()
+	defer arbiter.Unlock()
+	arbiter.meters[m] = struct{}{}
+	if !arbiter.started {
+		arbiter.started = true
+		go arbiter.tick()
+	}
+}
+
+func (m *StandardMeter) stopArbiter() {
+	arbiter.Lock()
+	defer arbiter.Unlock()
+	delete(arbiter.meters, m)
+	if len(arbiter.meters) == 0 && arbiter.started {
+		arbiter.cancel <- struct{}{}
+		arbiter.started = false
+	}
+}
+
 // meterArbiter ticks meters every 5s from a single goroutine.
 // meters are references in a set for future stopping.
 type meterArbiter struct {
 	sync.RWMutex
 	started bool
 	meters  map[*StandardMeter]struct{}
-	ticker  *time.Ticker
+	cancel  chan struct{}
 }
 
-var arbiter = meterArbiter{ticker: time.NewTicker(5e9), meters: make(map[*StandardMeter]struct{})}
+var arbiter = meterArbiter{
+	meters: make(map[*StandardMeter]struct{}),
+	cancel: make(chan struct{}),
+}
 
 // Ticks meters on the scheduled interval
 func (ma *meterArbiter) tick() {
+	ticker := time.NewTicker(5 * time.Second)
+	defer ticker.Stop()
 	for {
 		select {
-		case <-ma.ticker.C:
+		case <-ticker.C:
 			ma.tickMeters()
+		case <-ma.cancel:
+			return
 		}
 	}
 }

meter_test.go

@@ -29,7 +29,6 @@ func BenchmarkMeterParallel(b *testing.B) {
 func TestMeterConcurrency(t *testing.T) {
 	rand.Seed(time.Now().Unix())
 	ma := meterArbiter{
-		ticker: time.NewTicker(time.Millisecond),
 		meters: make(map[*StandardMeter]struct{}),
 	}
 	m := newStandardMeter()
@@ -62,7 +61,6 @@ func TestGetOrRegisterMeter(t *testing.T) {
 
 func TestMeterDecay(t *testing.T) {
 	ma := meterArbiter{
-		ticker: time.NewTicker(time.Millisecond),
 		meters: make(map[*StandardMeter]struct{}),
 	}
 	m := newStandardMeter()