debian-mirror-gitlab/workhorse/internal/filestore/reader.go
2021-02-22 17:27:13 +05:30

17 lines
255 B
Go

package filestore
import "io"
type hardLimitReader struct {
r io.Reader
n int64
}
func (h *hardLimitReader) Read(p []byte) (int, error) {
nRead, err := h.r.Read(p)
h.n -= int64(nRead)
if h.n < 0 {
err = ErrEntityTooLarge
}
return nRead, err
}