From ad55f6a867b5a1e0de50ea17d4dec8b99f285728 Mon Sep 17 00:00:00 2001 From: qydysky Date: Sun, 28 May 2023 17:06:04 +0800 Subject: [PATCH] fix --- file/FileWR.go | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/file/FileWR.go b/file/FileWR.go index a8c0289..8959087 100644 --- a/file/FileWR.go +++ b/file/FileWR.go @@ -383,21 +383,17 @@ func transferIO(r io.Reader, w io.Writer, byteInSec int64) (e error) { limit := l.New(1, "1s", "-1s") defer limit.Close() - buf := make([]byte, byteInSec) - for { - n, err := r.Read(buf) - if n != 0 { + for buf := make([]byte, byteInSec); true; { + if n, err := r.Read(buf); n != 0 { w.Write(buf[:n]) } else if err != nil { - e = err - break + return err } limit.TO() } - } else { - _, e = io.Copy(w, r) + } else if _, err := io.Copy(w, r); err != nil { + return err } - return nil } -- 2.39.2