fix: not stop when error occurred using pmux

nwtgck · Apr 29, 2024 · 166ef13 · 166ef13
1 parent 4ba2623
commit 166ef13
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/cmd/client/client.go b/cmd/client/client.go
@@ -305,7 +305,11 @@ func clientHandleWithPmux(ln net.Listener, httpClient *http.Client, headers []pi
 	for {
 		conn, err := ln.Accept()
 		if err != nil {
-			break
+			cmd.Vlog.Log(
+				fmt.Sprintf("error(accept): %v", errors.WithStack(err)),
+				fmt.Sprintf("error(accept): %+v", errors.WithStack(err)),
+			)
+			continue
 		}
 		stream, err := pmuxClient.Open()
 		if err != nil {

diff --git a/cmd/server/server.go b/cmd/server/server.go
@@ -290,7 +290,11 @@ func serverHandleWithPmux(httpClient *http.Client, headers []piping_util.KeyValu
 	for {
 		stream, err := pmuxServer.Accept()
 		if err != nil {
-			return err
+			cmd.Vlog.Log(
+				fmt.Sprintf("error(pmux accept): %v", errors.WithStack(err)),
+				fmt.Sprintf("error(pmux accept): %+v", errors.WithStack(err)),
+			)
+			continue
 		}
 		conn := dialLoop()
 		go func() {