缓存结束后增加Content-Length对比

internal/nodes/http_writer.go

@@ -9,6 +9,7 @@ import (
 	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
 	"github.com/TeaOSLab/EdgeNode/internal/utils"
 	"github.com/iwind/TeaGo/lists"
+	"github.com/iwind/TeaGo/types"
 	"net"
 	"net/http"
 	"strings"
@@ -223,17 +224,29 @@ func (this *HTTPWriter) Close() {
 	// cache writer
 	if this.cacheWriter != nil {
 		if this.isOk {
-			err := this.cacheWriter.Close()
+			// 对比Content-Length
-			if err == nil {
+			contentLengthString := this.Header().Get("Content-Length")
-				this.cacheStorage.AddToList(&caches.Item{
+			if len(contentLengthString) > 0 {
-					Type:       this.cacheWriter.ItemType(),
+				contentLength := types.Int64(contentLengthString)
-					Key:        this.cacheWriter.Key(),
+				if contentLength != this.cacheWriter.BodySize() {
-					ExpiredAt:  this.cacheWriter.ExpiredAt(),
+					this.isOk = false
-					HeaderSize: this.cacheWriter.HeaderSize(),
+					_ = this.cacheWriter.Discard()
-					BodySize:   this.cacheWriter.BodySize(),
+				}
-					Host:       this.req.Host,
+			}
-					ServerId:   this.req.Server.Id,
+
-				})
+			if this.isOk {
+				err := this.cacheWriter.Close()
+				if err == nil {
+					this.cacheStorage.AddToList(&caches.Item{
+						Type:       this.cacheWriter.ItemType(),
+						Key:        this.cacheWriter.Key(),
+						ExpiredAt:  this.cacheWriter.ExpiredAt(),
+						HeaderSize: this.cacheWriter.HeaderSize(),
+						BodySize:   this.cacheWriter.BodySize(),
+						Host:       this.req.Host,
+						ServerId:   this.req.Server.Id,
+					})
+				}
 			}
 		} else {
 			_ = this.cacheWriter.Discard()