37--go语言学习之日志配置logrus

go get github.com/sirupsen/logrus

package main

import (
	"fmt"
	"net/http"
	"os"

	"github.com/gin-gonic/gin"

	"github.com/sirupsen/logrus"
)

// 创建log
var log = logrus.New()

func initLog() error {
	// 设置日志格式  json
	log.Formatter = &logrus.JSONFormatter{}
	// 创建日志文件 ./aa.log:根路径文件名 os.O_CREATE:不存在则创建 os.O_WRONLY:可读可写 os.O_APPEND:追加  0644:权限
	file, err := os.OpenFile("./aa.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
	if err != nil {
		fmt.Println("打开日志文件失败", err)
		return err
	}
	// 绑定日志输出文件
	log.Out = file
	// 设置gin框架的发布版本
	gin.SetMode(gin.ReleaseMode)
	// 将gin框架日志也输出到本地配置日志
	gin.DefaultWriter = log.Out
	// 设置日志级别
	log.Level = logrus.InfoLevel
	return nil
}
func main() {
	err := initLog()
	if err != nil {
		fmt.Println("初始化日志失败", err.Error())
	}
	engine := gin.Default()
	engine.GET("log", func(c *gin.Context) {
		// 输出日志格式1
		log.WithFields(logrus.Fields{
			"url":        c.Request.URL,
			"method":     c.Request.Method,
			"remoteAddr": c.Request.RemoteAddr,
		}).Info()
		responseData := struct {
			Code int         `json:"code"`
			Msg  string      `json:"msg"`
			Data interface{} `json:"data"`
		}{http.StatusOK, "ok", nil}
		c.JSON(http.StatusOK, responseData)
		return
	})
	engine.Run(":9001")
}

go get github.com/lestrrat-go/file-rotatelogs
go get github.com/rifflock/lfshook

package main

import (
	"fmt"
	"net/http"
	"os"
	"path"
	"time"

	"github.com/rifflock/lfshook"

	rotatelogs "github.com/lestrrat-go/file-rotatelogs"

	"github.com/sirupsen/logrus"

	"github.com/gin-gonic/gin"
)

// 自定义日志变量
var (
	logFilePath = "./"     // 日志存储路径
	logFileName = "bb.log" // 日志文件名
)

func main() {
	engine := gin.Default()
	// 使用日志
	engine.Use(logMiddleware())
	engine.GET("log2", func(c *gin.Context) {
		responseData := struct {
			Code int         `json:"code"`
			Msg  string      `json:"msg"`
			Data interface{} `json:"data"`
		}{http.StatusOK, "ok", nil}
		c.JSON(http.StatusOK, responseData)
		return
	})
	engine.Run(":9002")
}

func logMiddleware() gin.HandlerFunc {
	// 定义日志文件
	fileName := path.Join(logFilePath, logFileName)
	// 创建日志文件
	file, err := os.OpenFile(fileName, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
	if err != nil {
		fmt.Println("打开日志文件失败", err)
		return nil
	}
	// 实例化
	logger := logrus.New()
	// 设置日志级别
	logger.SetLevel(logrus.DebugLevel)
	// 绑定日志输出文件
	logger.Out = file
	// 设置rotatelogs 绑定日志策略
	rotate, _ := rotatelogs.New(
		// 分割后的文件名称
		fileName+".%Y%m%d.log",
		// 设置软链  指向最新日志文件
		rotatelogs.WithLinkName(fileName),
		// 设置最大保存时间 (7天)
		rotatelogs.WithMaxAge(7*24*time.Hour),
		// 设置每次生成日志文件的时间间隔   (1天)
		rotatelogs.WithRotationTime(24*time.Hour),
	)
	// 设置hook  并绑定rotatelog
	writerMap := lfshook.WriterMap{
		logrus.InfoLevel:  rotate,
		logrus.FatalLevel: rotate,
		logrus.DebugLevel: rotate,
		logrus.WarnLevel:  rotate,
		logrus.ErrorLevel: rotate,
		logrus.PanicLevel: rotate,
	}
	// logrus添加hook
	logger.AddHook(lfshook.NewHook(writerMap, &logrus.JSONFormatter{
		TimestampFormat: "2006-02-02 02:02:02", // 必须配置， 其他下面的可删除
		//DisableTimestamp:  false,
		//DisableHTMLEscape: false,
		//DataKey:           "",
		//FieldMap:          nil,
		//CallerPrettyfier:  nil,
		//PrettyPrint:       false,
	}))
	return func(c *gin.Context) {
		c.Next()
		logger.WithFields(logrus.Fields{
			"url":        c.Request.URL,
			"method":     c.Request.Method,
			"remoteAddr": c.Request.RemoteAddr,
			"clientIp":   c.ClientIP(),
		}).Info()
	}
}