feat(smb): support range header (close #3192)

This commit is contained in:
Andy Hsu
2023-02-20 16:46:38 +08:00
parent 923937b530
commit 0d07d81802
3 changed files with 153 additions and 3 deletions

View File

@ -69,3 +69,25 @@ func (l limitWriter) Write(p []byte) (n int, err error) {
func LimitWriter(w io.Writer, size int64) io.Writer {
return &limitWriter{w: w, limit: size}
}
type ReadCloser struct {
io.Reader
io.Closer
}
type CloseFunc func() error
func (c CloseFunc) Close() error {
return c()
}
func NewReadCloser(reader io.Reader, close CloseFunc) io.ReadCloser {
return ReadCloser{
Reader: reader,
Closer: close,
}
}
func NewLimitReadCloser(reader io.Reader, close CloseFunc, limit int64) io.ReadCloser {
return NewReadCloser(io.LimitReader(reader, limit), close)
}