increase refcount before dropping dialsLk initially

511d7a82 · Jeromy · 41d999eb · 511d7a82
Commit 511d7a82 authored Oct 28, 2016 by Jeromy
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

dial_sync.go dial_sync.go +3 -1

No files found.
--- a/dial_sync.go
+++ b/dial_sync.go
@@ -36,7 +36,6 @@ type activeDial struct {
 }

 func (dr *activeDial) wait(ctx context.Context) (*Conn, error) {
-	dr.incref()
 	defer dr.decref()
 	select {
 	case <-dr.waitch:
@@ -97,6 +96,9 @@ func (ds *DialSync) getActiveDial(p peer.ID) *activeDial {
 		go actd.start(adctx)
 	}

+	// increase ref count before dropping dialsLk
+	actd.incref()
+
 	return actd
 }