add ctx canceled err check

libp2p · Nov 21, 2023 · 12fcd27 · 12fcd27
1 parent b63ad60
commit 12fcd27
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internal/net/message_manager.go b/internal/net/message_manager.go
@@ -295,8 +295,10 @@ func (ms *peerMessageSender) SendRequest(ctx context.Context, pmes *pb.Message)
 
 		mes := new(pb.Message)
 		if err := ms.ctxReadMsg(ctx, mes); err != nil {
-			_ = ms.s.Reset()
-			ms.s = nil
+			if err != context.Canceled {
+				_ = ms.s.Reset()
+				ms.s = nil
+			}
 
 			if retry {
 				logger.Debugw("error reading message", "error", err)