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{})