return canceled err not retry

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