TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-02 14:00:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Partial Content新创建时不对比Content-MD5

Browse Source
This commit is contained in:
GoEdgeLab
2024-05-08 10:30:47 +08:00
parent 8a46b87095
commit 9c79d5ff85
1 changed files with 16 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
internal/nodes/http_writer.go
View File

@@ -355,23 +355,28 @@ func (this *HTTPWriter) PrepareCache(resp *http.Response, size int64) {
}
this.cacheWriter = cacheWriter
// 对比Content-MD5
{
if this.isPartial {
partialWriter, ok := cacheWriter.(*caches.PartialFileWriter)
if ok {
if partialWriter.Ranges().Version >= 2 && partialWriter.Ranges().ContentMD5 != this.Header().Get("Content-MD5") {
_ = this.cacheWriter.Discard()
this.cacheWriter = nil
return
// 判断是否新创建的缓存文件
this.partialFileIsNew = partialWriter.IsNew()
if this.partialFileIsNew {
var contentMD5 = this.rawWriter.Header().Get("Content-MD5")
if len(contentMD5) > 0 {
partialWriter.SetContentMD5(contentMD5)
}
} else {
// 对比Content-MD5
if partialWriter.Ranges().Version >= 2 && partialWriter.Ranges().ContentMD5 != this.Header().Get("Content-MD5") {
_ = this.cacheWriter.Discard()
this.cacheWriter = nil
return
}
}
}
}
// 判断是否新创建的缓存文件
if this.isPartial {
this.partialFileIsNew = cacheWriter.(*caches.PartialFileWriter).IsNew()
}
// 写入Header
var headerBuf = utils.SharedBufferPool.Get()
for k, v := range this.Header() {
@@ -416,11 +421,6 @@ func (this *HTTPWriter) PrepareCache(resp *http.Response, size int64) {
return
}
partialWriter.SetBodyLength(total)
var contentMD5 = this.rawWriter.Header().Get("Content-MD5")
if len(contentMD5) > 0 {
partialWriter.SetContentMD5(contentMD5)
}
}
var filterReader = readers.NewFilterReaderCloser(resp.Body)
this.cacheIsFinished = true
@@ -480,11 +480,6 @@ func (this *HTTPWriter) PrepareCache(resp *http.Response, size int64) {
// 写入total
if !writtenTotal && total > 0 {
partialWriter.SetBodyLength(total)
var contentMD5 = this.rawWriter.Header().Get("Content-MD5")
if len(contentMD5) > 0 {
partialWriter.SetContentMD5(contentMD5)
}
writtenTotal = true
}