feat(webdav): support http chunked request (close #5161 in #5162)

But we do not recommend not adding the content-length header when putting files
AlistGo · Sep 5, 2023 · b604e21 · b604e21
1 parent 3c66db9
commit b604e21
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/internal/stream/stream.go b/internal/stream/stream.go
@@ -26,6 +26,16 @@ type FileStream struct {
 	peekBuff *bytes.Reader
 }
 
+func (f *FileStream) GetSize() int64 {
+	if f.tmpFile != nil {
+		info, err := f.tmpFile.Stat()
+		if err == nil {
+			return info.Size()
+		}
+	}
+	return f.Obj.GetSize()
+}
+
 func (f *FileStream) GetMimetype() string {
 	return f.Mimetype
 }

diff --git a/pkg/utils/file.go b/pkg/utils/file.go
@@ -126,7 +126,7 @@ func CreateTempFile(r io.Reader, size int64) (*os.File, error) {
 		_ = os.Remove(f.Name())
 		return nil, errs.NewErr(err, "CreateTempFile failed")
 	}
-	if size != 0 && readBytes != size {
+	if size > 0 && readBytes != size {
 		_ = os.Remove(f.Name())
 		return nil, errs.NewErr(err, "CreateTempFile failed, incoming stream actual size= %d, expect = %d ", readBytes, size)
 	}