mirror of
https://github.com/get-drexa/drive.git
synced 2025-12-01 14:01:40 +00:00
25 lines
457 B
Go
25 lines
457 B
Go
package ioext
|
|
|
|
import "io"
|
|
|
|
type LimitReadCloser struct {
|
|
reader io.ReadCloser
|
|
limitReader io.Reader
|
|
}
|
|
|
|
func NewLimitReadCloser(reader io.ReadCloser, length int64) *LimitReadCloser {
|
|
return &LimitReadCloser{
|
|
reader: reader,
|
|
limitReader: io.LimitReader(reader, length),
|
|
}
|
|
}
|
|
|
|
func (r *LimitReadCloser) Read(p []byte) (n int, err error) {
|
|
return r.limitReader.Read(p)
|
|
}
|
|
|
|
func (r *LimitReadCloser) Close() error {
|
|
return r.reader.Close()
|
|
}
|
|
|