Skip to content

Commit 461e546

Browse files
committed
Fi download bug
1 parent 40d2336 commit 461e546

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

routers/repo/download.go

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,10 @@ import (
2222
// ServeData download file from io.Reader
2323
func ServeData(ctx *context.Context, name string, reader io.Reader) error {
2424
buf := make([]byte, 1024)
25-
n, _ := reader.Read(buf)
25+
n, err := reader.Read(buf)
26+
if err != nil && err != io.EOF {
27+
return err
28+
}
2629
if n >= 0 {
2730
buf = buf[:n]
2831
}
@@ -46,7 +49,7 @@ func ServeData(ctx *context.Context, name string, reader io.Reader) error {
4649
ctx.Resp.Header().Set("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, name))
4750
}
4851

49-
_, err := ctx.Resp.Write(buf)
52+
_, err = ctx.Resp.Write(buf)
5053
if err != nil {
5154
return err
5255
}

0 commit comments

Comments
 (0)