理解底层— —Golang的log库，二开实现自定义Logger

Bertha 。 2024-03-24 22:08 75阅读 0赞

## 理解底层— —Golang的log库，实现自定义Logger ##

### 1 分析实现思路 ###

> 基于golang中自带的log库实现：对日志实现设置日志级别，每天生成一个文件，同时添加上前缀以及展示文件名等

*  日志级别，通过添加prefix：\[INFO\]、\[DEBUG\]等来实现
 *  每天生成一个日志文件：写日志之前判断当前时间是否为新的一天
 *  日志文件命名：获取每天的时间来实现命名，同时添加读写锁保证并发安全
 *  获取调用日志文件代码行数及文件名：runtime.Caller获取函数调用栈

### 2 实战 ###

#### 2.1 server.go ####

package main
    
    import "myTest/inter/log_pro/logger"
    
    func main() {
        
    	logger.SetFile("/Users/xsky/GolandProjects/MyTest/inter/log_pro/log/demo.log")
    	logger.SetLevel(0)
    
    	logger.Debug("hello %s", "ziyi")
    	logger.Info("hello %s", "ziyi")
    }

#### 2.2 logger.go ####

package logger
    
    import (
    	"log"
    	"os"
    	"runtime"
    	"strconv"
    	"strings"
    	"sync"
    	"time"
    )
    
    //基于log库自定义实现logger
    var (
    	infoLogger  *log.Logger
    	debugLogger *log.Logger
    
    	logOut     *os.File
    	logLevel   int
    	currentDay int //每天生成一个日志文件
    	logFile    string
    	fileLock   sync.RWMutex //读写锁，保证同一时间只有一个协程重命名文件
    )
    
    const (
    	DebugLevel = iota //0
    	InfoLevel         //1
    )
    
    func SetLevel(level int) {
        
    	logLevel = level
    }
    
    func init() {
        
    	fileLock = sync.RWMutex{
        }
    }
    
    func SetFile(file string) {
        
    	var err error
    	logOut, err = os.OpenFile(file, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0664)
    	if err != nil {
        
    		panic(err)
    	} else {
        
    		//初始化自定义的Logger（基于golang中的log库）
    		//log.LstdFlags表示时间格式等
    		//log.Llongfile表示文件名及调用代码的位置,log.Llongfile=》改为通过getCallTrace获取前缀
    		currentDay = time.Now().YearDay()
    		infoLogger = log.New(logOut, "[INFO] ", log.LstdFlags)
    		debugLogger = log.New(logOut, "[DEBUG] ", log.LstdFlags)
    		logFile = file
    	}
    }
    
    func checkIfDayChange() {
        
    	fileLock.Lock()
    	defer fileLock.Unlock()
    	day := time.Now().YearDay()
    	if day == currentDay {
        
    		return
    	} else {
        
    		//关闭之前的文件，重命名，并生成新的文件
    		logOut.Close()
    		postFix := time.Now().Add(-24 * time.Hour).Format("20060102")
    		err := os.Rename(logFile, logFile+"."+postFix)
    		if err != nil {
        
    			//TODO 重命名日志文件失败，根据自身情况做处理
    		}
    		logOut, err = os.OpenFile(logFile, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0664)
    		if err != nil {
        
    			//TODO 打开新的日志文件失败，根据自己业务需求做处理
    		}
    		infoLogger = log.New(logOut, "[INFO] ", log.LstdFlags)
    		debugLogger = log.New(logOut, "[DEBUG] ", log.LstdFlags)
    		currentDay = day
    	}
    }
    
    //golang中的any相当于interface{}空接口
    func Debug(format string, v ...any) {
        
    	if logLevel <= DebugLevel {
        
    		checkIfDayChange()
    		debugLogger.Printf(getPrefix()+format, v)
    	}
    }
    
    func Info(format string, v ...any) {
        
    	if logLevel <= InfoLevel {
        
    		checkIfDayChange()
    		infoLogger.Printf(getPrefix()+format, v)
    	}
    }
    
    //获取函数调用栈关系：拿到调用Info或者Debug所在的文件名及代码行数（runtime包）
    func getCallTrace() (string, int) {
        
    	_, file, lineNo, ok := runtime.Caller(3)
    	if ok {
        
    		return file, lineNo
    	} else {
        
    		return "", 0
    	}
    }
    
    //获取调用Info、Debug代码所在行数，文件名只获取最后三级
    func getPrefix() string {
        
    	file, lineNo := getCallTrace()
    	path := strings.Split(file, "/")
    	if len(path) > 3 {
        
    		file = strings.Join(path[len(path)-3:], "/")
    	}
    	return file + ":" + strconv.Itoa(lineNo) + " "
    }

### 3 效果 ###

> 运行server.go查看效果：

![在这里插入图片描述][15f6f2f3251e4fe7b7bbfc8d97226d11.png]

目录结构：  
![在这里插入图片描述][0732661cb92a438680455e4a7b964ab4.png]

[15f6f2f3251e4fe7b7bbfc8d97226d11.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/13/db4ee63815844c6b9ccf103219c3ca5a.png
[0732661cb92a438680455e4a7b964ab4.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/13/eeb018f7ab9a49519950e97da3be40b7.png