feat: support read timeout not set

zeromake · Jul 20, 2022 · ade9d70 · ade9d70
1 parent 9144482
commit ade9d70
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/tty/hijack.go b/pkg/tty/hijack.go
@@ -158,9 +158,11 @@ func Copy(ctx context.Context, dst net.Conn, src io.Reader, readTimeout time.Dur
 		nr, er := src.Read(buf)
 		if nr > 0 {
 			// docker container is stop check
-			err = dst.SetReadDeadline(time.Now().Add(readTimeout))
-			if err != nil {
-				break
+			if readTimeout > 0 {
+				err = dst.SetReadDeadline(time.Now().Add(readTimeout))
+				if err != nil {
+					break
+				}
 			}
 			nw, ew := dst.Write(buf[0:nr])
 			if nw < 0 || nr < nw {