fasthttp + `page partial gziped cache`: 页面输出服务性能提升20% - ToB企服应用市场:ToB评测及商务社交产业平台

复制代码

复制代码

package main
import (
"bytes"
_ "embed"
"fmt"
"log"
"os"
"github.com/klauspost/compress/gzip"
"github.com/valyala/bytebufferpool"
"github.com/valyala/fasthttp"
)
//go:embed raw.html
var html string
//go:embed raw.js
var js string
func testGzipedHttp() {
topHalf, digest := gzip.GetGzipedData([]byte(html)) // cache 页面的前一半, digest 是这些内容的 crc32 的校验和
requestHandler := func(ctx *fasthttp.RequestCtx) {
ctx.Response.Header.Add("Content-Type", "text/plain")
ctx.Response.Header.Add("Content-Encoding", "gzip")
switch string(ctx.Request.URI().RequestURI()) {
case "/1": // direct output
w, _ := gzip.NewWriterLevel(ctx, gzip.BestCompression)
w.Write([]byte(html))
w.Write([]byte(js))
w.Close()
case "/2":
w := gzip.GetWriter(ctx) // 使用对象池
w.WriteHeader() // 写 gzip 的头部信息，10 字节
w.WriteGzipedData([]byte(html), topHalf, digest)
// 当这个缓存是第一次输出的时候，可以传入 digest 值，这样可以少算一次 crc32
// 当不是第一次输出的时候，第三个参数 digest 填 0
w.Write([]byte(js))
gzip.PutWriter(w) // 必须调用，写入尾部信息，并放回对象池
}
}
s := &fasthttp.Server{
Handler: requestHandler,
}
if err := s.ListenAndServe(":8080"); err != nil {
log.Fatalf("error in ListenAndServe: %v", err)
}
}
func main() {
testGzipedHttp()
}

复制代码