reset stream and return err when ctx canceled

libp2p · Nov 24, 2023 · ae0987e · ae0987e
1 parent 097cb9a
commit ae0987e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/internal/net/message_manager.go b/internal/net/message_manager.go
@@ -295,11 +295,12 @@ 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 {
+				// retry would be same error
 				return nil, err
 			}
-			_ = ms.s.Reset()
-			ms.s = nil
 			if retry {
 				logger.Debugw("error reading message", "error", err)
 				return nil, err