improve error handling of tcp proxy.

apache · Dec 12, 2024 · 676d5fe · 676d5fe
1 parent f5abeb8
commit 676d5fe
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/gocase/util/client.go b/tests/gocase/util/client.go
@@ -102,11 +102,17 @@ func SimpleTCPProxy(ctx context.Context, t testing.TB, to string, slowdown bool)
 						time.Sleep(time.Millisecond * 100)
 					}
 					n, err := src.Read(buffer)
-					if err != nil && !errors.Is(err, io.EOF) {
+					if err != nil {
+						if errors.Is(err, io.EOF) {
+							break COPY_LOOP
+						}
 						return err
 					}
 					_, err = dest.Write(buffer[:n])
-					if err != nil && !errors.Is(err, io.EOF) {
+					if err != nil {
+						if errors.Is(err, io.EOF) {
+							break COPY_LOOP
+						}
 						return err
 					}
 				}