Skip to content

Commit d82b370

Browse files
committed
Check File.Close error for get_http.go
1 parent da1332a commit d82b370

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

get_http.go

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -135,9 +135,14 @@ func (g *HttpGetter) GetFile(dst string, u *url.URL) error {
135135
if err != nil {
136136
return err
137137
}
138-
defer f.Close()
139138

140-
_, err = io.Copy(f, resp.Body)
139+
n, err := io.Copy(f, resp.Body)
140+
if err == nil && n < resp.ContentLength {
141+
err = io.ErrShortWrite
142+
}
143+
if err1 := f.Close(); err == nil {
144+
err = err1
145+
}
141146
return err
142147
}
143148

0 commit comments

Comments
 (0)