correct refcount

libp2p · Jan 18, 2021 · fb39f1f · fb39f1f
1 parent 3d54a98
commit fb39f1f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rtrefresh/rt_refresh_manager.go b/rtrefresh/rt_refresh_manager.go
@@ -104,7 +104,9 @@ func (r *RtRefreshManager) Close() error {
 // error and close. The channel is buffered and safe to ignore.
 func (r *RtRefreshManager) Refresh(force bool) <-chan error {
 	resp := make(chan error, 1)
+	r.refcount.Add(1)
 	go func() {
+		defer r.refcount.Done()
 		select {
 		case r.triggerRefresh <- &triggerRefreshReq{respCh: resp, forceCplRefresh: force}:
 		case <-r.ctx.Done():