一万年来谁著史，三千里外欲封侯。这篇文章主要讲述Go 每日一库之 gorilla/handlers相关的知识，希望能为你提供帮助。
简介 上一篇文章中，我们介绍了 gorilla web 开发工具包中的路由管理库gorilla/mux，在文章最后我们介绍了如何使用中间件处理通用的逻辑。在日常 Go Web 开发中，开发者遇到了很多相同的中间件需求，gorilla/handlers（后文简称为handlers）收集了一些比较常用的中间件。一起来看看吧~
关于中间件，前面几篇文章已经介绍的很多了。这里就不赘述了。handlers库提供的中间件可用于标准库net/http和所有支持http.Handler接口的框架。由于gorilla/mux也支持http.Handler接口，所以也可以与handlers库结合使用。这就是兼容标准的好处。
项目初始化& 安装 本文代码使用 Go Modules。
【Go 每日一库之 gorilla/handlers】创建目录并初始化：

$ mkdir gorilla/handlers & & cd gorilla/handlers $ go mod init github.com/darjun/go-daily-lib/gorilla/handlers

安装gorilla/handlers库：

$ go get -u github.com/gorilla/handlers

下面依次介绍各个中间件和相应的源码。
日志 handlers提供了两个日志中间件：

• LoggingHandler：以 Apache 的Common Log Format日志格式记录 HTTP 请求日志；
  
• CombinedLoggingHandler：以 Apache的Combined Log Format日志格式记录 HTTP 请求日志，Apache 和 nginx 默认都使用这种日志格式。
  

两种日志格式差别很小，Common Log Format格式如下：

%h %l %u %t "%r" %> s %b

各个指示符含义如下：

• %h：客户端的 IP 地址或主机名；
  
• %l：RFC 1413定义的客户端标识，由客户端机器上的identd程序生成。如果不存在，则该字段为-；
  
• %u：已验证的用户名。如果不存在，该字段为-；
  
• %t：时间，格式为day/month/year:hour:minute:second zone，其中：
  
• • day：2位数字；
    
  • month：月份缩写，3个字母，如Jan；
    
  • year：4位数字；
    
  • hour：2位数字；
    
  • minute：2位数字；
    
  • second：2位数字；
    
  • zone：+或-后跟4位数字；
    
  • 例如：21/Jul/2021:06:27:33 +0800
    
• %r：包含 HTTP 请求行信息，例GET /index.html HTTP/1.1；
  
• %> s：服务器发送给客户端的状态码，例如200；
  
• %b：响应长度（字节数）。
  

Combined Log Format格式如下：

%h %l %u %t "%r" %> s %b "%Refereri" "%User-Agenti"

可见相比Common Log Format只是多了：

• %Refereri：HTTP 首部中的Referer信息；
  
• %User-Agenti：HTTP 首部中的User-Agent信息。
  

对中间件，我们可以让它作用于全局，即全部处理器，也可以让它只对某些处理器生效。如果要对所有处理器生效，可以调用Use()方法。如果只需要作用于特定的处理器，在注册时用中间件将处理器包装一层：

func index(w http.ResponseWriter, r *http.Request) fmt.Fprintln(w, "Hello World")type greeting stringfunc (g greeting) ServeHTTP(w http.ResponseWriter, r *http.Request) fmt.Fprintf(w, "Welcome, %s", g)func main() r := mux.NewRouter() r.Handle("/", handlers.LoggingHandler(os.Stdout, http.HandlerFunc(index))) r.Handle("/greeting", handlers.CombinedLoggingHandler(os.Stdout, greeting("dj")))http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

上面代码中LoggingHandler只作用于处理函数index，CombinedLoggingHandler只作用于处理器greeting("dj")。
运行代码，通过浏览器访问localhost:8080和localhost:8080/greeting：

::1 - - [21/Jul/2021:06:39:45 +0800] "GET / HTTP/1.1" 200 12 ::1 - - [21/Jul/2021:06:39:54 +0800] "GET /greeting HTTP/1.1" 200 11 "" "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.164 Safari/537.36"

对照前面分析的指示符，很容易看出各个部分。
由于*mux.Router的Use()方法接受类型为MiddlewareFunc的中间件：

type MiddlewareFunc func(http.Handler) http.Handler

而handlers.LoggingHandler/CombinedLoggingHandler并不满足，所以还需要包装一层才能传给Use()方法：

func Logging(handler http.Handler) http.Handler return handlers.CombinedLoggingHandler(os.Stdout, handler)func main() r := mux.NewRouter() r.Use(Logging) r.HandleFunc("/", index) r.Handle("/greeting/", greeting("dj"))http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

另外handlers还提供了CustomLoggingHandler，我们可以利用它定义自己的日志中间件：

func CustomLoggingHandler(out io.Writer, h http.Handler, f LogFormatter) http.Handler

最关键的LogFormatter类型定义：

type LogFormatterParams struct Request*http.Request URLurl.URL TimeStamptime.Time StatusCode int Sizeinttype LogFormatter func(writer io.Writer, params LogFormatterParams)

我们实现一个简单的LogFormatter，记录时间 + 请求行 + 响应码：

func myLogFormatter(writer io.Writer, params handlers.LogFormatterParams) var buf bytes.Buffer buf.WriteString(time.Now().Format("2006-01-02 15:04:05 -0700")) buf.WriteString(fmt.Sprintf(` "%s %s %s" `, params.Request.Method, params.URL.Path, params.Request.Proto)) buf.WriteString(strconv.Itoa(params.StatusCode)) buf.WriteByte(\\n)writer.Write(buf.Bytes())func Logging(handler http.Handler) http.Handler return handlers.CustomLoggingHandler(os.Stdout, handler, myLogFormatter)

使用：

func main() r := mux.NewRouter() r.Use(Logging) r.HandleFunc("/", index) r.Handle("/greeting/", greeting("dj"))http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

现在记录的日志是下面这种格式：

2021-07-21 07:03:18 +0800 "GET /greeting/ HTTP/1.1" 200

翻看源码，我们可以发现LoggingHandler/CombinedLoggingHandler/CustomLoggingHandler都是基于底层的loggingHandler实现的，不同的是LoggingHandler使用了预定义的writeLog作为LogFormatter，CombinedLoggingHandler使用了预定义的writeCombinedLog作为LogFormatter，而CustomLoggingHandler使用我们自己定义的LogFormatter：

func CombinedLoggingHandler(out io.Writer, h http.Handler) http.Handler return loggingHandlerout, h, writeCombinedLogfunc LoggingHandler(out io.Writer, h http.Handler) http.Handler return loggingHandlerout, h, writeLogfunc CustomLoggingHandler(out io.Writer, h http.Handler, f LogFormatter) http.Handler return loggingHandlerout, h, f

预定义的writeLog/writeCombinedLog实现如下：

func writeLog(writer io.Writer, params LogFormatterParams) buf := buildCommonLogLine(params.Request, params.URL, params.TimeStamp, params.StatusCode, params.Size) buf = append(buf, \\n) writer.Write(buf)func writeCombinedLog(writer io.Writer, params LogFormatterParams) buf := buildCommonLogLine(params.Request, params.URL, params.TimeStamp, params.StatusCode, params.Size) buf = append(buf, ` "`...) buf = appendQuoted(buf, params.Request.Referer()) buf = append(buf, `" "`...) buf = appendQuoted(buf, params.Request.UserAgent()) buf = append(buf, ", \\n) writer.Write(buf)

它们都是基于buildCommonLogLine构造基本信息，writeCombinedLog还分别调用http.Request.Referer()和http.Request.UserAgent获取了Referer和User-Agent信息。
loggingHandler定义如下：

type loggingHandler struct writerio.Writer handlerhttp.Handler formatter LogFormatter

loggingHandler实现有一个比较巧妙的地方：为了记录响应码和响应大小，定义了一个类型responseLogger包装原来的http.ResponseWriter，在写入时记录信息：

type responseLogger struct whttp.ResponseWriter status int sizeintfunc (l *responseLogger) Write(b []byte) (int, error) size, err := l.w.Write(b) l.size += size return size, errfunc (l *responseLogger) WriteHeader(s int) l.w.WriteHeader(s) l.status = sfunc (l *responseLogger) Status() int return l.statusfunc (l *responseLogger) Size() int return l.size

loggingHandler的关键方法ServeHTTP()：

func (h loggingHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) t := time.Now() logger, w := makeLogger(w) url := *req.URLh.handler.ServeHTTP(w, req) if req.MultipartForm != nil req.MultipartForm.RemoveAll()params := LogFormatterParams Request:req, URL:url, TimeStamp:t, StatusCode: logger.Status(), Size:logger.Size(),h.formatter(h.writer, params)

构造LogFormatterParams对象，调用对应的LogFormatter函数。
压缩 如果客户端请求中有Accept-Encoding首部，服务器可以使用该首部指示的算法将响应压缩，以节省网络流量。handlers.CompressHandler中间件启用压缩功能。还有一个CompressHandlerLevel可以指定压缩级别。实际上CompressHandler就是使用gzip.DefaultCompression调用的CompressHandlerLevel：

func CompressHandler(h http.Handler) http.Handler return CompressHandlerLevel(h, gzip.DefaultCompression)

看代码：

func index(w http.ResponseWriter, r *http.Request) fmt.Fprintln(w, "Hello World")type greeting stringfunc (g greeting) ServeHTTP(w http.ResponseWriter, r *http.Request) fmt.Fprintf(w, "Welcome, %s", g)func main() r := mux.NewRouter() r.Use(handlers.CompressHandler) r.HandleFunc("/", index) r.Handle("/greeting/", greeting("dj"))http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

运行，请求localhost:8080，通过 Chrome 开发者工具的 Network 页签可以看到响应采用了 gzip 压缩：

文章图片
忽略一些细节处理，CompressHandlerLevel函数代码如下：

func CompressHandlerLevel(h http.Handler, level int) http.Handler const ( gzipEncoding= "gzip" flateEncoding = "deflate" )return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) var encoding string for _, curEnc := range strings.Split(r.Header.Get(acceptEncoding), ",") curEnc = strings.TrimSpace(curEnc) if curEnc == gzipEncoding || curEnc == flateEncoding encoding = curEnc breakif encoding == "" h.ServeHTTP(w, r) returnif r.Header.Get("Upgrade") != "" h.ServeHTTP(w, r) returnvar encWriter io.WriteCloser if encoding == gzipEncoding encWriter, _ = gzip.NewWriterLevel(w, level) else if encoding == flateEncoding encWriter, _ = flate.NewWriter(w, level)defer encWriter.Close()w.Header().Set("Content-Encoding", encoding) r.Header.Del(acceptEncoding)cw := & compressResponseWriter w:w, compressor: encWriter,w = httpsnoop.Wrap(w, httpsnoop.Hooks Write: func(httpsnoop.WriteFunc) httpsnoop.WriteFunc return cw.Write , WriteHeader: func(httpsnoop.WriteHeaderFunc) httpsnoop.WriteHeaderFunc return cw.WriteHeader , Flush: func(httpsnoop.FlushFunc) httpsnoop.FlushFunc return cw.Flush , ReadFrom: func(rff httpsnoop.ReadFromFunc) httpsnoop.ReadFromFunc return cw.ReadFrom , )h.ServeHTTP(w, r) )

从请求Accept-Encoding首部中获取客户端指示的压缩算法。如果客户端未指定，或请求首部中有Upgrade，则不压缩。反之，则压缩。根据识别的压缩算法，创建对应gzip或flate的io.Writer实现对象。
与前面的日志中间件一样，为了压缩写入的内容，新增类型compressResponseWriter封装http.ResponseWriter，重写Write()方法，将写入的字节流传入前面创建的io.Writer实现压缩：

type compressResponseWriter struct compressor io.Writer whttp.ResponseWriterfunc (cw *compressResponseWriter) Write(b []byte) (int, error) h := cw.w.Header() if h.Get("Content-Type") == "" h.Set("Content-Type", http.DetectContentType(b))h.Del("Content-Length")return cw.compressor.Write(b)

内容类型 我们可以通过handler.ContentTypeHandler指定请求的Content-Type必须在我们给出的类型中，只对POST/PUT/PATCH方法生效。例如我们限制登录请求必须通过application/x-www-form-urlencoded的形式发送：

func main() r := mux.NewRouter() r.HandleFunc("/", index) r.Methods("GET").Path("/login").HandlerFunc(login) r.Methods("POST").Path("/login"). Handler(handlers.ContentTypeHandler(http.HandlerFunc(dologin), "application/x-www-form-urlencoded"))http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

这样，只要请求/login的Content-Type不是application/x-www-form-urlencoded就会返回 415 错误。我们可以故意写错，再请求看看表现：

Unsupported content type "application/x-www-form-urlencoded"; expected one of ["application/x-www-from-urlencoded"]

ContentTypeHandler的实现非常简单：

func ContentTypeHandler(h http.Handler, contentTypes ...string) http.Handler return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) if !(r.Method == "PUT" || r.Method == "POST" || r.Method == "PATCH") h.ServeHTTP(w, r) returnfor _, ct := range contentTypes if isContentType(r.Header, ct) h.ServeHTTP(w, r) returnhttp.Error(w, fmt.Sprintf("Unsupported content type %q; expected one of %q", r.Header.Get("Content-Type"), contentTypes), http.StatusUnsupportedMediaType) )

就是读取Content-Type首部，判断是否在我们指定的类型中。
方法分发器 在上面的例子中，我们注册路径/login的GET和POST方法处理采用r.Methods("GET").Path("/login").HandlerFunc(login)这种冗长的写法。handlers.MethodHandler可以简化这种写法：

func main() r := mux.NewRouter() r.HandleFunc("/", index) r.Handle("/login", handlers.MethodHandler "GET":http.HandlerFunc(login), "POST": http.HandlerFunc(dologin), )http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

MethodHandler底层是一个map[string]http.Handler类型，它的ServeHTTP()方法根据请求的 Method 调用不同的处理：

type MethodHandler map[string]http.Handlerfunc (h MethodHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) if handler, ok := h[req.Method]; ok handler.ServeHTTP(w, req) else allow := []string for k := range h allow = append(allow, k)sort.Strings(allow) w.Header().Set("Allow", strings.Join(allow, ", ")) if req.Method == "OPTIONS" w.WriteHeader(http.StatusOK) else http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)

方法如果未注册，则返回405 Method Not Allowed。有一个方法除外，OPTIONS。该方法通过Allow首部返回支持哪些方法。
重定向 handlers.CanonicalHost可以将请求重定向到指定的域名，同时指定重定向响应码。在同一个服务器对应多个域名时比较有用：

func index(w http.ResponseWriter, r *http.Request) fmt.Fprintln(w, "hello world")func main() r := mux.NewRouter() r.Use(handlers.CanonicalHost("http://www.gorillatoolkit.org", 302)) r.HandleFunc("/", index) http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

上面将所有请求以 302 重定向到http://www.gorillatoolkit.org。
CanonicalHost的实现也很简单：

func CanonicalHost(domain string, code int) func(h http.Handler) http.Handler fn := func(h http.Handler) http.Handler return canonicalh, domain, codereturn fn

关键类型canonical：

type canonical struct hhttp.Handler domain string codeint

核心方法：

func (c canonical) ServeHTTP(w http.ResponseWriter, r *http.Request) dest, err := url.Parse(c.domain) if err != nil c.h.ServeHTTP(w, r) returnif dest.Scheme == "" || dest.Host == "" c.h.ServeHTTP(w, r) returnif !strings.EqualFold(cleanHost(r.Host), dest.Host) dest := dest.Scheme + "://" + dest.Host + r.URL.Path if r.URL.RawQuery != "" dest += "?" + r.URL.RawQueryhttp.Redirect(w, r, dest, c.code) returnc.h.ServeHTTP(w, r)

由源码可知，域名不合法或未指定协议（Scheme）或域名（Host）的请求下不转发。
Recovery 之前我们自己实现了PanicRecover中间件，避免请求处理时 panic。handlers提供了一个RecoveryHandler可以直接使用：

func PANIC(w http.ResponseWriter, r *http.Request) panic(errors.New("unexpected error"))func main() r := mux.NewRouter() r.Use(handlers.RecoveryHandler(handlers.PrintRecoveryStack(true))) r.HandleFunc("/", PANIC) http.Handle("/", r) log.Fatal(http.ListenAndServe(":8080", nil))

选项PrintRecoveryStack表示 panic 时输出堆栈信息。
RecoveryHandler的实现与之前我们自己编写的基本一样：

type recoveryHandler struct handlerhttp.Handler loggerRecoveryHandlerLogger printStack boolfunc (h recoveryHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) defer func() if err := recover(); err != nil w.WriteHeader(http.StatusInternalServerError) h.log(err)()h.handler.ServeHTTP(w, req)

总结 GitHub 上有很多开源的 Go Web 中间件实现，可以直接拿来使用，避免重复造轮子。handlers很轻量，容易与标准库net/http和 gorilla 路由库mux结合使用。
大家如果发现好玩、好用的 Go 语言库，欢迎到 Go 每日一库 GitHub 上提交 issue????
参考

1. gorilla/handlers GitHub：github.com/gorilla/handlers
   
2. Go 每日一库 GitHub：https://github.com/darjun/go-daily-lib
   

我 我的博客：https://darjun.github.io
欢迎关注我的微信公众号【GoUpUp】，共同学习，一起进步~

文章图片

推荐阅读

• 

  口碑比较好的家具品牌 哪个品牌家具好
• 

  恒大御景半岛酒店多少钱一晚 恒大御景半岛酒店电话
• 

  如何查看redis全部链接 redis查看哪个是主节点
• 

  mysql 加锁 sql语句怎么写 mysql数据表加锁
• 

  CAD中的dra是什么命令怎么用
• 

  有什么好用的精华液推荐？
• 

  雷锋精神伴我行
• 

  王者荣耀游走装备调整 王者荣耀体验服游走装备效果恩赐规则改动一览
• 

  联想2400pro硒鼓清零
• 

  php链接数据库测试代码 php连接数据库的函数
• 

  怎样判断句子是举例子的，如何判断举例子的说明方法
• 

  液晶电视显示屏折了
• 

  拍摄游戏的软件
• 

  Xshell怎么卸载？Xshell卸载方法介绍
• 

  沉鱼落雁十二生肖是哪一肖 沉鱼落雁什么意思
• 

  新房地基开工祝福语
• 

  晕车人可以开车吗 晕车人能开车吗
• 

  鱼为什么睁着眼睛睡觉 因为鱼是没有眼睑
• 

  抖音就这样爱你爱你爱你随时都要一起是什么歌
• 

  美国三大体育联赛

• 北亚数据恢复raid损坏导致硬盘灯报警离线存储无法使用的数据恢复案例
• Arch Linux 安装 Anbox
• 类加载的时机
• 关于String对象到底占多少字节（）
• HotSpot 虚拟机对象探秘
• MYSQL索引优化法则
• JVM 内存结构
• 10个显示彩色字符的批处理代码
• JVM-03