fix(stream): http chucked upload issue (#1152)

* fix(stream): http chucked upload issue

* fix(stream): use MmapThreshold

* fix(stream): improve caching mechanism and handle size=0 case

* fix bug

* fix(buffer): optimize ReadAt method for improved performance

* fix(upload): handle Content-Length and File-Size headers for better size management

* fix(189pc): 移除重复限速

* fix(upload): handle negative file size during streaming uploads

* fix(upload): update header key from File-Size to X-File-Size for size retrieval

---------

Co-authored-by: j2rong4cn <j2rong@qq.com>
This commit is contained in:
TwoOnefour
2025-09-15 19:36:16 +08:00
committed by GitHub
parent c1d03c5bcc
commit cbbb5ad231
8 changed files with 207 additions and 45 deletions

View File

@ -8,83 +8,86 @@ import (
// 用于存储不复用的[]byte
type Reader struct {
bufs [][]byte
length int
offset int
size int64
offset int64
}
func (r *Reader) Len() int {
return r.length
func (r *Reader) Size() int64 {
return r.size
}
func (r *Reader) Append(buf []byte) {
r.length += len(buf)
r.size += int64(len(buf))
r.bufs = append(r.bufs, buf)
}
func (r *Reader) Read(p []byte) (int, error) {
n, err := r.ReadAt(p, int64(r.offset))
n, err := r.ReadAt(p, r.offset)
if n > 0 {
r.offset += n
r.offset += int64(n)
}
return n, err
}
func (r *Reader) ReadAt(p []byte, off int64) (int, error) {
if off < 0 || off >= int64(r.length) {
if off < 0 || off >= r.size {
return 0, io.EOF
}
n, length := 0, int64(0)
n := 0
readFrom := false
for _, buf := range r.bufs {
newLength := length + int64(len(buf))
if readFrom {
w := copy(p[n:], buf)
n += w
} else if off < newLength {
nn := copy(p[n:], buf)
n += nn
if n == len(p) {
return n, nil
}
} else if newOff := off - int64(len(buf)); newOff >= 0 {
off = newOff
} else {
nn := copy(p, buf[off:])
if nn == len(p) {
return nn, nil
}
n += nn
readFrom = true
w := copy(p[n:], buf[int(off-length):])
n += w
}
if n == len(p) {
return n, nil
}
length = newLength
}
return n, io.EOF
}
func (r *Reader) Seek(offset int64, whence int) (int64, error) {
var abs int
switch whence {
case io.SeekStart:
abs = int(offset)
case io.SeekCurrent:
abs = r.offset + int(offset)
offset = r.offset + offset
case io.SeekEnd:
abs = r.length + int(offset)
offset = r.size + offset
default:
return 0, errors.New("Seek: invalid whence")
}
if abs < 0 || abs > r.length {
if offset < 0 || offset > r.size {
return 0, errors.New("Seek: invalid offset")
}
r.offset = abs
return int64(abs), nil
r.offset = offset
return offset, nil
}
func (r *Reader) Reset() {
clear(r.bufs)
r.bufs = nil
r.length = 0
r.size = 0
r.offset = 0
}
func NewReader(buf ...[]byte) *Reader {
b := &Reader{}
b := &Reader{
bufs: make([][]byte, 0, len(buf)),
}
for _, b1 := range buf {
b.Append(b1)
}