Go Web编程添加服务器错误和访问日志

目录

• 前言
• 初始化日志记录器
• 添加错误日志
• 添加访问日志

// ./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)
}

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
   ...
}

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])
}

type ResponseWriter interface {
  Header() Header
  Write([]byte) (int, error)
  WriteHeader(statusCode int)
}

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
}

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/router.go
func RegisterRoutes(r *mux.Router) {
  ...
  // apply Logging middleware
  r.Use(middleware.Logging(), middleware.AccessLogging)
  ...
}

• Django日志logging的配置和自定义添加方式
• 如何在django中添加日志功能
• Python Django框架实现应用添加logging日志操作示例
• GO日志打印如何添加goroutineid

大家感兴趣的内容

• 1Goland激活码破解永久版及安装详细教程(亲测可以)
• 2Go语言string,int,int64 ,float之间类
• 3Go语言中的Array、Slice、Map和Set使用详解
• 4Go语言的GOPATH与工作目录详解
• 5Go语言转换所有字符串为大写或者小写的方法
• 6Go语言interface详解
• 7五步让你成为GO 语言高手
• 8Go语言命令行操作命令详细介绍
• 9Go语言编程中字符串切割方法小结
• 10我放弃Python转Go语言的9大理由(附优秀书籍推荐)

最近更新的内容

• 浅析Go中序列化与反序列化的基本使用
• Go GORM版本2.0新特性介绍
• Go语言实现优雅关机和重启的示例详解
• golang recover函数使用中的一些坑解析
• Go语言实战之实现一个简单分布式系统
• Go语言for range(按照键值循环)遍历操作
• 基于Go语言实现简单网络聊天室(命令行模式)
• 详解golang中模板的常用语法
• 从源码深入理解golang RWMutex读写锁操作
• Go语言反射reflect.Value实现方法的调用

常用在线小工具