lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-09-19 13:47:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: drains http body if buffer is too small

Browse Source 
Signed-off-by: divinerapier <poriter.coco@gmail.com>
This commit is contained in:
divinerapier
2019-12-26 11:28:43 +08:00
parent 84640d07b7
commit 4cbb6fa199
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
weed/util/http_util.go
View File

@@ -235,8 +235,16 @@ func ReadUrl(fileUrl string, offset int64, size int, buf []byte, isReadRange boo
if err != nil { if err != nil {
return n, err return n, err
} }
if n == int64(len(buf)) {
break
}
} }
// drains the response body to avoid memory leak
data, err := ioutil.ReadAll(reader)
if len(data) != 0 {
err = fmt.Errorf("buffer size is too small. remains %d", len(data))
}
return n, err
} }
func ReadUrlAsStream(fileUrl string, offset int64, size int, fn func(data []byte)) (int64, error) { func ReadUrlAsStream(fileUrl string, offset int64, size int, fn func(data []byte)) (int64, error) {