[better_conn_close_error] c1

internal/transport/http2_client.go

@@ -59,7 +59,6 @@ type http2Client struct {
 	ctx        context.Context
 	cancel     context.CancelFunc
 	ctxDone    <-chan struct{} // Cache the ctx.Done() chan.
-	closeErr   error           // Reason the transport is closed. Only set if ctx is canceled.
 	userAgent  string
 	md         metadata.MD
 	conn       net.Conn // underlying communication channel
@@ -110,6 +109,7 @@ type http2Client struct {
 
 	mu            sync.Mutex // guard the following variables
 	state         transportState
+	closeErr      error // Reason the transport is closed. Only set if ctx is canceled.
 	activeStreams map[uint32]*Stream
 	// prevGoAway ID records the Last-Stream-ID in the previous GOAway frame.
 	prevGoAwayID uint32

internal/transport/transport_test.go

@@ -781,6 +781,12 @@ func (s) TestGracefulClose(t *testing.T) {
 			defer wg.Done()
 			str, err := ct.NewStream(ctx, &CallHdr{})
 			if err != nil {
+				// The NewStream calls race with closing of the transport (after
+				// the last stream is closed).
+				// - if it is after the transport is closed (case <-ct.ctxDone),
+				//   we don't care about the error.
+				// - if it's before the transport is closed (default), we want
+				//   the NewStream call to succeed.
 				select {
 				case <-ct.ctxDone:
 				default: