use ErrStreamReset for resetting streams

libp2p · Nov 20, 2024 · 39abe7e · 39abe7e
1 parent 3eaea39
commit 39abe7e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/session.go b/session.go
@@ -325,10 +325,14 @@ func (s *Session) close(shutdownErr error, sendGoAway bool, errCode uint32) erro
 	<-s.sendDoneCh
 	<-s.recvDoneCh
 
+	resetErr := shutdownErr
+	if _, ok := resetErr.(*GoAwayError); !ok {
+		resetErr = fmt.Errorf("%w: connection closed: %w", ErrStreamReset, shutdownErr)
+	}
 	s.streamLock.Lock()
 	defer s.streamLock.Unlock()
 	for id, stream := range s.streams {
-		stream.forceClose(s.shutdownErr)
+		stream.forceClose(resetErr)
 		delete(s.streams, id)
 		stream.memorySpan.Done()
 	}