fix err on conn close

const.go

@@ -173,7 +173,7 @@ const (
 	// It's not an implementation choice, the value defined in the specification.
 	initialStreamWindow = 256 * 1024
 	maxStreamWindow     = 16 * 1024 * 1024
-	goAwayWaitTime      = 5 * time.Second
+	goAwayWaitTime      = 50 * time.Millisecond
 )
 
 const (

session.go

@@ -334,7 +334,7 @@ func (s *Session) close(shutdownErr error, sendGoAway bool, errCode uint32) erro
 	s.streamLock.Lock()
 	defer s.streamLock.Unlock()
 	for id, stream := range s.streams {
-		stream.forceClose()
+		stream.forceClose(fmt.Errorf("%w: connection closed: %w", ErrStreamReset, s.shutdownErr))
 		delete(s.streams, id)
 		stream.memorySpan.Done()
 	}

stream.go

@@ -42,7 +42,7 @@ type Stream struct {
 	state                 streamState
 	writeState, readState halfStreamState
 	stateLock             sync.Mutex
-	resetErr              *StreamError
+	resetErr              error
 
 	recvBuf segmentedBuffer
 
@@ -365,7 +365,7 @@ func (s *Stream) Close() error {
 }
 
 // forceClose is used for when the session is exiting
-func (s *Stream) forceClose() {
+func (s *Stream) forceClose(err error) {
 	s.stateLock.Lock()
 	if s.readState == halfOpen {
 		s.readState = halfReset
@@ -374,6 +374,7 @@ func (s *Stream) forceClose() {
 		s.writeState = halfReset
 	}
 	s.state = streamFinished
+	s.resetErr = err
 	s.notifyWaiting()
 	s.stateLock.Unlock()