with panic recover

spirt-t · Apr 13, 2021 · 34e7431 · 34e7431
1 parent 964f1a9
commit 34e7431
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/transport/http2_client.go b/internal/transport/http2_client.go
@@ -387,6 +387,15 @@ func newHTTP2Client(connectCtx, ctx context.Context, addr resolver.Address, opts
 		return nil, err
 	}
 	go func() {
+		defer func() {
+			if r := recover(); r != nil {
+				if err != nil {
+					err = fmt.Errorf("%w; panic %v", err, r)
+				} else {
+					err = fmt.Errorf("panic %v", r)
+				}
+			}
+		}()
 		t.loopy = newLoopyWriter(clientSide, t.framer, t.controlBuf, t.bdpEst)
 		err := t.loopy.run()
 		if err != nil {