Fix leaked goroutine

grpc · Mar 11, 2024 · 8af4b0b · 8af4b0b
1 parent 809175c
commit 8af4b0b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/transport/http2_client.go b/internal/transport/http2_client.go
@@ -1008,11 +1008,12 @@ func (t *http2Client) Close(err error) {
 	if err1 != nil {
 		t.logger.Infof("Failed to put goaway to server: %v\n", err1)
 	}
-	timer := time.NewTimer(10 * time.Second)
+	timer := time.NewTimer(2 * time.Second)
 	defer timer.Stop()
 	select {
 	case <-t.goAwaySent.Done():
 	case <-timer.C:
+	case <-t.ctx.Done():
 	}
 	t.controlBuf.setError(ErrConnClosing)
 	t.cancel()