1. package initialize
    3. import (
    4.     "encoding/json"
    5.     "github.com/tidwall/gjson"
    6.     "github.com/tidwall/sjson"
    7.     "go.uber.org/zap"
    8.     "go.uber.org/zap/zapcore"
    9.     "golang.org/x/net/context"
    10.     "gopkg.in/natefinch/lumberjack.v2"
    11.     "notification_service/global"
    12.     "os"
    13.     "time"
    14. )
    16. func InitLogger() {
    17.     development := os.Getenv("DEVELOPMENT")
    19.     if development == "1" { // 如果是开发环境
    20.         initDevelopmentLogger()
    21.     } else {
    22.         initProductionLogger()
    23.     }
    24. }
    26. func initDevelopmentLogger() {
    27.     config := zap.NewDevelopmentConfig()
    29.     logger, err := config.Build(zap.AddCaller())
    30.     if err != nil {
    31.         panic(err)
    32.     }
    34.     zap.ReplaceGlobals(logger)
    35. }
    37. func initProductionLogger() {
    38.     encoderConfig := zap.NewProductionEncoderConfig()     // 获取编码器
    39.     encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder // 使用 ISO-8601 时间格式
    40.     encoderConfig.TimeKey = "time"
    41.     encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder // 日志级别记为大写
    42.     encoder := zapcore.NewJSONEncoder(encoderConfig)        // NewJSONEncoder：json 格式输出。如果是 NewConsoleEncoder，则以字符串格式输出
    44.     infoLogLevel := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool { // 该日志等级函数记录 INFO 及以上级别的日志
    45.         return lvl >= zap.InfoLevel
    46.     })
    47.     errorLogLevel := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool { // 该日志等级函数记录 ERROR 及以上级别的日志
    48.         return lvl >= zap.ErrorLevel
    49.     })
    51.     infoFileSyncer := zapcore.AddSync(&lumberjack.Logger{
    52.         Filename:   "./log/info.log", // 日志文件存放目录，如果目录不存在会自动创建
    53.         MaxSize:    50,               // 文件大小限制，单位：MB
    54.         MaxBackups: 200,              // 最大保留日志文件数量
    55.         MaxAge:     180,              // 日志文件保留天数
    56.         Compress:   false,            // 是否压缩处理
    57.     })
    58.     errorFileSyncer := zapcore.AddSync(&lumberjack.Logger{
    59.         Filename:   "./log/error.log",
    60.         MaxSize:    50,
    61.         MaxBackups: 200,
    62.         MaxAge:     180,
    63.         Compress:   false,
    64.     })
    65.     redisSyncer := zapcore.AddSync(&RedisLogger{}) // 日志存入 Redis
    67.     infoCore := zapcore.NewCore(encoder, zapcore.NewMultiWriteSyncer(infoFileSyncer, redisSyncer), infoLogLevel)
    68.     errorCore := zapcore.NewCore(encoder, errorFileSyncer, errorLogLevel)
    70.     logger := zap.New(zapcore.NewTee(infoCore, errorCore), zap.AddCaller()) // zap.AddCaller()：显示文件名和行号
    72.     zap.ReplaceGlobals(logger) // 替换全局 logger
    73. }
    75. type RedisLogger struct{} // 需要实现 io.Writer 接口
    77. func (*RedisLogger) Write(p []byte) (n int, err error) { // 实现 io.Writer 接口即实现 Write 方法
    78.     log := string(p)
    80.     if gjson.Get(log, "trace_id").String() == "" { // 如果日志中不包含 trace_id
    81.         return
    82.     }
    84.     var redisLog string
    85.     if gjson.Get(log, "log_type").String() == "access_log" {
    86.         type AccessLog struct {
    87.             TraceID          string   `json:"trace_id,omitempty"`
    88.             Time             string   `json:"time,omitempty"`
    89.             Timestamp        int64    `json:"timestamp,omitempty"`
    90.             LogType          string   `json:"log_type,omitempty"`
    91.             ClientID         string   `json:"client_id,omitempty"`
    92.             RequestURI       string   `json:"request_uri,omitempty"`
    93.             RequestMethod    string   `json:"request_method,omitempty"`
    94.             RequestHeaders   string   `json:"request_headers,omitempty"`
    95.             RequestBody      string   `json:"request_body,omitempty"`
    96.             ResponseBody     string   `json:"response_body,omitempty"`
    97.             NotificationType string   `json:"notification_type,omitempty"`
    98.             Receivers        []string `json:"receivers,omitempty"`
    99.         }
    100.         accessLog := AccessLog{}
    101.         _ = json.Unmarshal(p, &accessLog)
    102.         t, _ := time.ParseInLocation("2006-01-02T15:04:05+0800", accessLog.Time, time.Local)
    103.         accessLog.Time = ""
    104.         accessLog.Timestamp = t.UnixMilli()
    105.         redisLogBytes, _ := json.Marshal(accessLog)
    106.         redisLog = string(redisLogBytes)
    107.     } else {
    108.         redisLog, _ = sjson.Set(redisLog, "trace_id", gjson.Get(log, "trace_id").String())
    109.         t, _ := time.ParseInLocation("2006-01-02T15:04:05+0800", gjson.Get(log, "time").String(), time.Local)
    110.         redisLog, _ = sjson.Set(redisLog, "timestamp", t.UnixMilli())
    111.         redisLog, _ = sjson.Set(redisLog, "log_type", "exec_log")
    112.         log, _ = sjson.Delete(log, "level")
    113.         log, _ = sjson.Delete(log, "time")
    114.         log, _ = sjson.Delete(log, "caller")
    115.         log, _ = sjson.Delete(log, "msg")
    116.         log, _ = sjson.Delete(log, "trace_id")
    117.         redisLog, _ = sjson.Set(redisLog, "log_content", log)
    118.     }
    119.     global.Redis.RPush(context.Background(), "notification_service_log", redisLog)
    121.     return
    122. }