From 580a818b48798344c6f2ece64ba67150e3f3ae0a Mon Sep 17 00:00:00 2001
From: vyzo <vyzo@hackzen.org>
Date: Wed, 31 Mar 2021 18:30:30 +0300
Subject: [PATCH] merge dial contexts where possible

---
 swarm_dial.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/swarm_dial.go b/swarm_dial.go
index 2d3255b..513ce1c 100644
--- a/swarm_dial.go
+++ b/swarm_dial.go
@@ -320,6 +320,7 @@ func (s *Swarm) dialWorkerLoop(ctx context.Context, p peer.ID, reqch <-chan Dial
 		conn     *Conn
 		err      error
 		requests []int
+		dialed   bool
 	}
 
 	reqno := 0
@@ -454,6 +455,9 @@ loop:
 			requests[reqno] = pr
 
 			for _, ad := range tojoin {
+				if !ad.dialed {
+					ad.ctx = s.mergeDialContexts(ad.ctx, req.Ctx)
+				}
 				ad.requests = append(ad.requests, reqno)
 			}
 
@@ -490,6 +494,7 @@ loop:
 					continue
 				}
 
+				ad.dialed = true
 				dialed = true
 				last = i
 				active++
@@ -581,6 +586,18 @@ func (s *Swarm) addrsForDial(ctx context.Context, p peer.ID) ([]ma.Multiaddr, er
 	return goodAddrs, nil
 }
 
+func (s *Swarm) mergeDialContexts(a, b context.Context) context.Context {
+	dialCtx := a
+
+	if simConnect, reason := network.GetSimultaneousConnect(b); simConnect {
+		if simConnect, _ := network.GetSimultaneousConnect(a); !simConnect {
+			dialCtx = network.WithSimultaneousConnect(dialCtx, reason)
+		}
+	}
+
+	return dialCtx
+}
+
 func (s *Swarm) dialNextAddr(ctx context.Context, p peer.ID, addr ma.Multiaddr, resch chan dialResult) error {
 	// check the dial backoff
 	if forceDirect, _ := network.GetForceDirectDial(ctx); !forceDirect {
-- 
GitLab