Go-Logger: 一个简单的带日志级别处理的日志库

下面我将介绍一个简单的Go日志库实现，支持不同日志级别、日志输出格式化和基本日志记录功能。

核心功能

1. 支持多种日志级别：DEBUG, INFO, WARN, ERROR, FATAL
2. 可自定义日志输出格式
3. 支持输出到控制台和文件
4. 线程安全

实现代码

package logger

import (
	"fmt"
	"io"
	"log"
	"os"
	"sync"
	"time"
)

// 日志级别类型
type LogLevel int

const (
	DEBUG LogLevel = iota
	INFO
	WARN
	ERROR
	FATAL
)

var levelNames = []string{
	"DEBUG",
	"INFO",
	"WARN",
	"ERROR",
	"FATAL",
}

// Logger 结构体
type Logger struct {
	level      LogLevel
	logger     *log.Logger
	mu         sync.Mutex
	output     io.Writer
	timeFormat string
}

// NewLogger 创建一个新的Logger实例
func NewLogger(level LogLevel, output io.Writer) *Logger {
	return &Logger{
		level:      level,
		logger:     log.New(output, "", 0),
		output:     output,
		timeFormat: "2006-01-02 15:04:05",
	}
}

// SetLevel 设置日志级别
func (l *Logger) SetLevel(level LogLevel) {
	l.mu.Lock()
	defer l.mu.Unlock()
	l.level = level
}

// SetTimeFormat 设置时间格式
func (l *Logger) SetTimeFormat(format string) {
	l.mu.Lock()
	defer l.mu.Unlock()
	l.timeFormat = format
}

// 通用日志记录方法
func (l *Logger) log(level LogLevel, format string, args ...interface{}) {
	if level < l.level {
		return
	}

	l.mu.Lock()
	defer l.mu.Unlock()

	msg := fmt.Sprintf(format, args...)
	timestamp := time.Now().Format(l.timeFormat)
	levelName := levelNames[level]

	logMsg := fmt.Sprintf("[%s] [%s] %s", timestamp, levelName, msg)
	l.logger.Println(logMsg)

	if level == FATAL {
		os.Exit(1)
	}
}

// Debug 记录DEBUG级别日志
func (l *Logger) Debug(format string, args ...interface{}) {
	l.log(DEBUG, format, args...)
}

// Info 记录INFO级别日志
func (l *Logger) Info(format string, args ...interface{}) {
	l.log(INFO, format, args...)
}

// Warn 记录WARN级别日志
func (l *Logger) Warn(format string, args ...interface{}) {
	l.log(WARN, format, args...)
}

// Error 记录ERROR级别日志
func (l *Logger) Error(format string, args ...interface{}) {
	l.log(ERROR, format, args...)
}

// Fatal 记录FATAL级别日志并退出程序
func (l *Logger) Fatal(format string, args ...interface{}) {
	l.log(FATAL, format, args...)
}

使用示例

package main

import (
	"os"
	"./logger" // 替换为你的logger包路径
)

func main() {
	// 创建输出到控制台的logger
	consoleLogger := logger.NewLogger(logger.DEBUG, os.Stdout)
	
	// 创建输出到文件的logger
	file, err := os.OpenFile("app.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
	if err != nil {
		consoleLogger.Error("Failed to open log file: %v", err)
		return
	}
	defer file.Close()
	
	fileLogger := logger.NewLogger(logger.INFO, file)
	
	// 设置不同的日志级别
	consoleLogger.SetLevel(logger.DEBUG)  // 控制台输出所有级别
	fileLogger.SetLevel(logger.INFO)     // 文件只记录INFO及以上级别
	
	// 记录日志
	consoleLogger.Debug("This is a debug message")
	consoleLogger.Info("This is an info message")
	consoleLogger.Warn("This is a warning message")
	consoleLogger.Error("This is an error message")
	
	fileLogger.Info("This will be written to file")
	fileLogger.Debug("This debug message won't appear in file")
	
	// 修改时间格式
	consoleLogger.SetTimeFormat("2006/01/02 15:04:05.000")
	consoleLogger.Info("Log with custom time format")
	
	// Fatal日志会退出程序
	// consoleLogger.Fatal("This is a fatal error, program will exit")
}

扩展功能建议

1. 日志轮转：可以添加日志文件大小检查，实现日志轮转功能
2. JSON格式：支持JSON格式的日志输出
3. 多输出：支持同时输出到多个目标(控制台+文件+网络等)
4. 上下文信息：支持添加请求ID等上下文信息
5. 性能优化：使用缓冲写入提高性能

总结

这个简单的go-logger实现提供了基本的日志功能，包括：

• 多级别日志记录
• 线程安全
• 可自定义输出格式
• 支持多种输出目标

你可以根据需要进一步扩展功能，比如添加日志轮转、支持JSON格式输出等高级特性。