Go Web编程--给本身写的服务器添加错误和访问日志

错误日志和访问日志是一个服务器必须支持的功能，咱们教程里使用的服务器到目前为止尚未这两个功能。正好前两天也写了篇介绍logrus日志库的文章，那么今天的文章里就给咱们本身写的服务器加上错误日志和访问日志的功能。在介绍添加访问日志的时候会介绍一种经过编写中间件获取HTTP响应的StausCode和Body的方法。

Go Web 编程系列的每篇文章的源代码都打了对应版本的软件包，供你们参考。公众号中回复 gohttp11获取本文源代码

初始化日志记录器

咱们先来作一下初始化工做，在项目里初始化记录错误日志和访问日志的记录器Logger。

// ./utils/vlog
package vlog

import (
    "github.com/sirupsen/logrus"
    "os"
)

var ErrorLog *logrus.Logger
var AccessLog *logrus.Logger
var errorLogFile = "./tmp/log/error.log"
var accessLogFile = "./tmp/log/access.log"
func init () {
    initErrorLog()
    initAccessLog()
}

func initErrorLog() {
    ErrorLog = logrus.New()
    ErrorLog.SetFormatter(&logrus.JSONFormatter{})
    file , err := os.OpenFile(errorLogFile, os.O_RDWR | os.O_CREATE | os.O_APPEND, 0755)
    if err != nil {
        panic(err)
    }
    ErrorLog.SetOutput(file)
}

func initAccessLog() {
    AccessLog = logrus.New()
    AccessLog.SetFormatter(&logrus.JSONFormatter{})
    file , err := os.OpenFile(accessLogFile, os.O_RDWR | os.O_CREATE | os.O_APPEND, 0755)
    if err != nil {
        panic(err)
    }
    AccessLog.SetOutput(file)
}

• 咱们新定义一个package在init函数中来初始化记录器，这样服务器成功启动前就会初始化好记录器。
• /tmp/log这个目录要提早建立好，执行init函数时会自动建立好access.log和error.log。

添加错误日志

咱们建立服务器使用的net/http包的Server类型中，有一个ErrorLog字段供开发者设置记录错误日志用的记录器Logger，默认使用的是log包默认的记录器（应该是系统的标准错误）：

type Server struct {
   Addr    string  // TCP address to listen on, ":http" if empty
   Handler Handler // handler to invoke, http.DefaultServeMux if nil
   ...
   // ErrorLog specifies an optional logger for errors accepting
   // connections, unexpected behavior from handlers, and
   // underlying FileSystem errors.
   // If nil, logging is done via the log package's standard logger.
   ErrorLog *log.Logger
     ...
}

咱们以前在建立服务器的时候本身实现了Server类型的对象，那么如今要作的就是将上面初始化好的错误日志的记录器指定给Server的ErrorLog字段。

func main() {
  ...
  // 将`logrus`的Logger转换为io.Writer
    errorWriter := vlog.ErrorLog.Writer()
  // 记得关闭io.Writer
    defer errorWriter.Close()

    server := &http.Server{
        Addr:    ":8080",
        Handler: muxRouter,
    // 用记录器转换成的io.Writer建立log.Logger
        ErrorLog: log.New(vlog.ErrorLog.Writer(), "", 0),
    }
    ...
}

添加好错误日志的记录器后，咱们找个路由处理函数，在里面故意制造运行时错误验证一下是否能记录到错误。

func (*HelloHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
   ints := []int{0, 1, 2}
   fmt.Fprintf(w, "%v", ints[0:5])
}

在上面处理函数中，经过切片表达式越界故意制造了一个运行时错误，打开error.log后能看到文件里已经记录到这个运行时错误及其Stack trace

添加访问日志

和Server对象能够设置错误日志的记录器不同，访问日志只能是咱们经过本身编写中间件的方式来实现了。在记录访问日志的中间件里咱们会记录ip，method，path，query，request_body，status和response_body这些个字段的内容。

status和response_body两个字段来自请求对应的响应。响应在net/http包里是用http.ResponseWriter接口表示的

type ResponseWriter interface {
    Header() Header

    Write([]byte) (int, error)

    WriteHeader(statusCode int)
}

接口自己以及net/http提供的实现都没有让咱们进行读取的方法，因此在编写的用于记录访问日志的中间件里须要对net/http库自己实现的ResponseWriter作一层包装。

利用Go语言结构体类型嵌套匿名类型后，结构体拥有了被嵌套类型的全部导出字段和方法的特性，咱们能够很方便地对原来的ResponseWriter作一层包装，而后只从新实现须要更改的方法便可：

type ResponseWithRecorder struct {
   http.ResponseWriter
   statusCode int
   body bytes.Buffer
}

func (rec *ResponseWithRecorder) WriteHeader(statusCode int) {
   rec.ResponseWriter.WriteHeader(statusCode)
   rec.statusCode = statusCode
}

func (rec *ResponseWithRecorder) Write(d []byte) (n int, err error) {
   n, err = rec.ResponseWriter.Write(d)
   if err != nil {
      return
   }
   rec.body.Write(d)

   return
}

定义好新的类型后咱们从新实现了WriteHeader和Write方法，在向原来的ReponseWriter中写入后也会向ResponseWriteRecoder.statusCode和ResponseWriteRecoder.body写入对应的数据。这样咱们就能够在中间件里经过这两个字段访问响应码和响应数据了。

记录访问日志的中间件定义以下：

func AccessLogging (f http.Handler) http.Handler {

    // 建立一个新的handler包装http.HandlerFunc
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        buf := new(bytes.Buffer)
        buf.ReadFrom(r.Body)
        logEntry := vlog.AccessLog.WithFields(logrus.Fields{
            "ip": r.RemoteAddr,
            "method": r.Method,
            "path": r.RequestURI,
            "query": r.URL.RawQuery,
            "request_body": buf.String(),

        })

        wc := &ResponseWithRecorder{
            ResponseWriter: w,
            statusCode: http.StatusOK,
            body: bytes.Buffer{},
        }

        // 调用下一个中间件或者最终的handler处理程序
        f.ServeHTTP(wc, r)

        defer logEntry.WithFields(logrus.Fields{
            "status": wc.statusCode,
            "response_body": wc.body.String(),
        }).Info()

    })
}

在Router上应用建立好的AccessLogging中间件后，就能够正常的记录服务器的访问日志了。

// router/router.go
func RegisterRoutes(r *mux.Router) {
    ...
    // apply Logging middleware
    r.Use(middleware.Logging(), middleware.AccessLogging)
    ...
}

不过有两点须要注意一下

• 这里为了演示获取响应数据记录了response_body字段，若是是接口响应内容记录下还能够，可是若是是HTML仍是不记录的为好。
• 初始化ResponseWithRecorder时默认设置了statusCode时由于，服务器正确返回响应时不会显式调用WriteHeader方法，只有在返回NOT_FOUND之类的错误的时候才会调用WriteHeader方法，针对这种状况须要在初始化的时候把statusCode的默认值设置为200。

如今再访问服务器后打开access.log会看到刚刚的访问日志，就能看到刚刚请求的url，method，客户端IP等信息了。

{"ip":"......","level":"info","method":"GET","msg":"","path":"/index/","query":"","request_body":"","response_body":"Hello World1","status":200,"time":"2020-03-26T04:21:46Z"}

注意：文章只为说明演示方便，获取IP的方法没法获取代理后的真实IP，请悉知。

前文回顾

深刻学习用Go编写HTTP服务器

Go Web 编程--超详细的模板库应用指南

Go Web编程--使用Go语言建立静态文件服务器

Go Web编程--SecureCookie实现客户端Session管理

关于如何收集，标准化和集中化处理Golang日志的一些建议