缓存文件实现Sendfile

internal/caches/storage_file.go

@@ -819,6 +819,14 @@ func (this *FileStorage) IgnoreKey(key string) {
 	this.ignoreKeys.Push(key)
 }
 
+// CanSendfile 是否支持Sendfile
+func (this *FileStorage) CanSendfile() bool {
+	if this.options == nil {
+		return false
+	}
+	return this.options.EnableSendfile
+}
+
 // 绝对路径
 func (this *FileStorage) dir() string {
 	return this.options.Dir + "/p" + strconv.FormatInt(this.policy.Id, 10) + "/"

internal/caches/storage_interface.go

@@ -51,4 +51,7 @@ type StorageInterface interface {
 
 	// IgnoreKey 忽略某个Key，即不缓存某个Key
 	IgnoreKey(key string)
+
+	// CanSendfile 是否支持Sendfile
+	CanSendfile() bool
 }

internal/caches/storage_memory.go

@@ -347,6 +347,11 @@ func (this *MemoryStorage) IgnoreKey(key string) {
 	this.ignoreKeys.Push(key)
 }
 
+// CanSendfile 是否支持Sendfile
+func (this *MemoryStorage) CanSendfile() bool {
+	return false
+}
+
 // 计算Key Hash
 func (this *MemoryStorage) hash(key string) uint64 {
 	return xxhash.Sum64String(key)

internal/nodes/http_request_cache.go

@@ -540,7 +540,15 @@ func (this *HTTPRequest) doCacheRead(useStale bool) (shouldStop bool) {
 			this.writer.Prepare(resp, fileSize, reader.Status(), false)
 			this.writer.WriteHeader(reader.Status())
 
+			if storage.CanSendfile() {
+				if fp, canSendFile := this.writer.canSendfile(); canSendFile {
+					this.writer.sentBodyBytes, err = io.CopyBuffer(this.writer.rawWriter, fp, buf)
+				} else {
 					_, err = io.CopyBuffer(this.writer, resp.Body, buf)
+				}
+			} else {
+				_, err = io.CopyBuffer(this.writer, resp.Body, buf)
+			}
 			if err == io.EOF {
 				err = nil
 			}

internal/nodes/http_writer.go

@@ -1058,8 +1058,10 @@ func (this *HTTPWriter) Close() {
 		}
 	}
 
+	if this.sentBodyBytes == 0 {
 		this.sentBodyBytes = this.counterWriter.TotalBytes()
 	}
+}
 
 // Hijack Hijack
 func (this *HTTPWriter) Hijack() (conn net.Conn, buf *bufio.ReadWriter, err error) {

internal/nodes/http_writer_ext.go

@@ -0,0 +1,9 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build !plus
+// +build !plus
+
+package nodes
+
+func (this *HTTPWriter) canSendfile() (*os.File, bool) {
+	return nil, false
+}

internal/utils/writers/writer_bytes_counter.go

@@ -13,6 +13,10 @@ func NewBytesCounterWriter(rawWriter io.Writer) *BytesCounterWriter {
 	return &BytesCounterWriter{writer: rawWriter}
 }
 
+func (this *BytesCounterWriter) RawWriter() io.Writer {
+	return this.writer
+}
+
 func (this *BytesCounterWriter) Write(p []byte) (n int, err error) {
 	n, err = this.writer.Write(p)
 	this.count += int64(n)