fasthttp 中如何使用 linux 系统调用 `sendfile`

作者:张富春(ahfuzhang)，转载时请注明作者和引用链接，谢谢！

• cnblogs博客
• zhihu
• Github
• 公众号:一本正经的瞎扯
  
  

---

接上一篇：fasthttp 中如何使用Transfer-Encoding: chunked 方式的流式内容输出

首先是找了一个库，封装了 linux 下的 sendfile 系统调用：github.com/hslam/sendfile

上代码：

• go.mod 中：

replace (
	github.com/valyala/fasthttp => github.com/ahfuzhang/compress v1.49.2
)

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"io"
	syslog "log"
	"net"
	"os"
	"strconv"
	"time"

    "github.com/hslam/sendfile"
	"github.com/valyala/bytebufferpool"
	"github.com/valyala/fasthttp"
)

var fileToSend *os.File
var fileSize int64

func FasthttpHandler(ctx *fasthttp.RequestCtx) {
    ctx.HijackSetNoResponse(true)  // 不使用  fasthttp 框架的输出
	//ctx.Hijack(func(c net.Conn) {  // 这里会  go 出来一个协程执行  hijack 回调函数。 我改了一个版本，不配置就不会 go 出来协程
	//	log.Println("do nothing")
	//})
	//
	buf := bytebufferpool.Get()
	defer bytebufferpool.Put(buf)
	buf.Reset()
	buf.B = append(buf.B, "HTTP/1.1 200 OK\r\n"+
		"Server: fasthttp\r\n"+
		"Date: Thu, 26 Oct 2023 01:09:23 GMT\r\n"+
		"Content-Type: text/plain\r\n"+
		"Cache-Control: no-cache, no-store, must-revalidate\r\n"+
		"Transfer-Encoding: chunked\r\n"+
		"\r\n"...)
	conn := ctx.Conn()
	_, _ = conn.Write(buf.B)  // 写  http 头
    //
    for i := 0; i < 9; i++ {
           _, _ = conn.Write([]byte(fmt.Sprintf("%x\r\n%d \r\n", 1, i)))
	}
    //
    _, _ = conn.Write([]byte(fmt.Sprintf("%x\r\n", fileSize, i)))
    n, err := sendfile.SendFile(ctx.Conn(), int(fileToSend.Fd()), 0, fileSize)  // 使用系统调用
	if err != nil {
		syslog.Fatalln(err)
	}
    _, _ = conn.Write([]byte("\r\n"))
	syslog.Println("send count:", n)
    // 写入结束信息
    _, _ = conn.Write([]byte("0\r\n\r\n")
}

func main() {
     fileToSend, _ = os.Open("xxxx.file")
     fileSize = 1024*2
    //
	syslog.SetFlags(syslog.Lshortfile | syslog.LstdFlags)
	server := &fasthttp.Server{
		Handler: FasthttpHandler,
	}
	syslog.Fatalln(server.ListenAndServe(":8089"))
}

相比直接从内存中 send 数据，实际测试中并未发现明显的性能提升，也可能是我只使用了 2kb 的数据，差异不明显。

希望对大家有用， have fun