fix leaking goroutine

cenkalti · Mar 2, 2019 · b165cdd · b165cdd
1 parent 17515e1
commit b165cdd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/announcer/stop.go b/internal/announcer/stop.go
@@ -40,8 +40,11 @@ func (a *StopAnnouncer) Run() {
 
 	ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(a.timeout))
 	go func() {
-		<-a.closeC
-		cancel()
+		select {
+		case <-ctx.Done():
+		case <-a.closeC:
+			cancel()
+		}
 	}()
 
 	doneC := make(chan struct{})