Golang如何实现静态文件压缩与缓存_Golang Web静态文件优化方法实践

package main

import (
    "compress/gzip"
    "io"
    "net/http"
    "strings"
)

// gzipMiddleware 支持gzip压缩的中间件
func gzipMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 判断客户端是否支持gzip
        if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
            next.ServeHTTP(w, r)
            return
        }

        // 创建gzip writer
        gz := gzip.NewWriter(w)
        defer gz.Close()

        // 包装ResponseWriter，替换Write方法
        w.Header().Set("Content-Encoding", "gzip")
        w.Header().Del("Content-Length") // 压缩后长度变化，需删除原长度头

        gw := gzipResponseWriter{Writer: gz, ResponseWriter: w}
        next.ServeHTTP(gw, r)
    })
}

// 包装http.ResponseWriter以拦截Write调用
type gzipResponseWriter struct {
    io.Writer
    http.ResponseWriter
}

func (w gzipResponseWriter) Write(b []byte) (int, error) {
    return w.Writer.Write(b)
}

fs := http.FileServer(http.Dir("static/"))
http.Handle("/static/", gzipMiddleware(fs))

func cacheControlMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if strings.HasSuffix(r.URL.Path, ".js") || 
           strings.HasSuffix(r.URL.Path, ".css") {
            w.Header().Set("Cache-Control", "public, max-age=31536000, immutable")
        } else if strings.HasSuffix(r.URL.Path, ".png") || 
                  strings.HasSuffix(r.URL.Path, ".jpg") {
            w.Header().Set("Cache-Control", "public, max-age=604800")
        }
        next.ServeHTTP(w, r)
    })
}

hash := sha256.Sum256([]byte(fmt.Sprintf("%s-%d", filepath, info.ModTime().Unix())))
etag := fmt.Sprintf("\"%x\"", hash[:10])

// assets.go
var AssetMap = map[string]string{
    "main.js": "main.abcd1234.js",
}

// 模板中使用