ReadOrErr 读取文件后，关闭流 (#134)

gookit · Oct 13, 2023 · d10983d · d10983d
1 parent 081b02c
commit d10983d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/fsutil/opread.go b/fsutil/opread.go
@@ -85,6 +85,14 @@ func MustRead(in any) []byte {
 // ReadOrErr read contents from path or io.Reader, will panic on in type error
 func ReadOrErr(in any) ([]byte, error) {
 	r, err := NewIOReader(in)
+	defer func() {
+		if r != nil {
+			file, ok := r.(*os.File)
+			if ok {
+				err = file.Close()
+			}
+		}
+	}()
 	if err != nil {
 		return nil, err
 	}