Logrus 集成 color 库实现自定义日志颜色输出字符原理

本文主要是介绍Logrus 集成 color 库实现自定义日志颜色输出字符原理,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

问题背景

下列代码实现了使用 Logurs 日志框架输出日志时根据级别不同,使用对应的自定义颜色进行输出。那么思考下代码的逻辑是怎么实现的呢?

效果如下:

在这里插入图片描述

代码如下:

import ("fmt""github.com/sirupsen/logrus""github.com/fatih/color"
)func main() {logger := logrus.New()logger.Formatter = &CustomTextFormatter{ForceColors:   true,ColorInfo:     color.New(color.FgBlue),ColorWarning:  color.New(color.FgYellow),ColorError:    color.New(color.FgRed),ColorCritical: color.New(color.BgRed, color.FgWhite),}logger.Info("This is an info message")   // 输出蓝色的信息日志logger.Warn("This is a warning message") // 输出黄色的警告日志logger.Error("This is an error message") // 输出红色的错误日志logger.Fatal("This is a critical message") // 输出带有红色背景和白色文本的严重日志logger.Panic("This is a critical message with panic") // 输出带有红色背景和白色文本的严重日志,并引发 panic
}// 自定义格式化器,继承自 logrus.TextFormatter
type CustomTextFormatter struct {logrus.TextFormatterForceColors   boolColorInfo     *color.ColorColorWarning  *color.ColorColorError    *color.ColorColorCritical *color.Color
}// 格式化方法,用于将日志条目格式化为字节数组
func (f *CustomTextFormatter) Format(entry *logrus.Entry) ([]byte, error) {if f.ForceColors {switch entry.Level {case logrus.InfoLevel:f.ColorInfo.Println(entry.Message) // 使用蓝色打印信息日志case logrus.WarnLevel:f.ColorWarning.Println(entry.Message) // 使用黄色打印警告日志case logrus.ErrorLevel:f.ColorError.Println(entry.Message) // 使用红色打印错误日志case logrus.FatalLevel, logrus.PanicLevel:f.ColorCritical.Println(entry.Message) // 使用带有红色背景和白色文本的样式打印严重日志default:f.PrintColored(entry)}return nil, nil} else {return f.TextFormatter.Format(entry)}
}// 自定义方法,用于将日志条目以带颜色的方式打印出来
func (f *CustomTextFormatter) PrintColored(entry *logrus.Entry) {levelColor := color.New(color.FgCyan, color.Bold) // 定义蓝色和粗体样式levelText := levelColor.Sprintf("%-6s", entry.Level.String()) // 格式化日志级别文本msg := levelText + " " + entry.Messageif entry.HasCaller() {msg += " (" + entry.Caller.File + ":" + strconv.Itoa(entry.Caller.Line) + ")"  // 添加调用者信息}fmt.Fprintln(color.Output, msg) // 使用有颜色的方式打印消息到终端
}

了解转义序列的颜色格式字符串

假如我们在代码中执行以下代码,那么你将会在控制台中看到带有红色的 “Hello World”

str := fmt.Sprintf("\033[31m%s\033[0m", "Hello World")
fmt.Println(str)

这是因为我们使用 fmt.Sprintf 方法生成了一个带有转义序列的颜色格式字符串。

在这个格式字符串中,%s 是一个占位符,表示将要插入的字符串。而 \033[31m\033[0m 是转义序列,用于在终端中设置文本的颜色,\033[31m 用于设置文本的颜色为红色,\033[0m 则是用于重置文本的颜色为默认值。

通过将实际的字符串插入到 %s 的位置,然后打印输出这个格式化字符串,可以在终端中实现相应部分带有颜色显示的效果。

**注:**该格式字符串的使用依赖于终端对 ANSI 转义序列的支持,因此可能在某些环境或终端下无效。

调用链

首先,我们的代码中定义了一个继承 logrus.TextFormatter 的结构体 CustomTextFormatter,并实现Format方法来自定义日志的输出格式。还将 Logrus 实例的 Formatter 设置为自定义格式化器 CustomTextFormatter

logger := logrus.New()
logger.Formatter = &CustomTextFormatter{ForceColors:   true,ColorInfo:     color.New(color.FgBlue),ColorWarning:  color.New(color.FgYellow),ColorError:    color.New(color.FgRed),ColorCritical: color.New(color.BgRed, color.FgWhite),
}logger.Info("This is an info message")
logger.Warn("This is a warning message")
logger.Error("This is an error message")
logger.Fatal("This is a critical message")

当调用 logger.Info() 时,会调用到 Logger.Log() 方法输出日志。Logger.Log() 首先通过调用 logger.IsLevelEnabled(level) 方法检查指定的日志级别是否启用。如果未启用,则直接返回,不执行后续的日志记录操作。

如果指定的日志级别是启用的,接下来会创建一个新的日志条目 entry,通过调用 logger.newEntry() 方法实现。

然后,通过调用 entry.Log(level, args...) 方法,将日志级别和参数传递给日志条目进行记录。这表示具体的日志记录操作和格式化输出由 entry.Log 方法负责执行。

最后,通过调用 logger.releaseEntry(entry) 方法释放和回收该日志条目的资源。

func (logger *Logger) Log(level Level, args ...interface{}) {if logger.IsLevelEnabled(level) {entry := logger.newEntry()entry.Log(level, args...)logger.releaseEntry(entry)}
}

entry.Log 中,首先使用 entry.Dup() 方法复制当前 Entry 实例的 LoggerTimeContexterr 创建一个新的 Entry 实例,并返回该新实例的指针,并将新实例的时间设置为当前时间,级别和消息设置为传入的参数值。这个新实例被称为 newEntry

后续,通过调用 newEntry.write() 方法,将新实例 newEntry 的内容写入缓冲区 buffer 中。

最后,释放新实例 newEntry 的缓冲区,检查日志级别是否需要触发 panic,并在需要触发 panic 时抛出新实例 newEntry

func (entry *Entry) log(level Level, msg string) {var buffer *bytes.BuffernewEntry := entry.Dup()if newEntry.Time.IsZero() {newEntry.Time = time.Now()}newEntry.Level = levelnewEntry.Message = msgnewEntry.Logger.mu.Lock()reportCaller := newEntry.Logger.ReportCallerbufPool := newEntry.getBufferPool()newEntry.Logger.mu.Unlock()if reportCaller {newEntry.Caller = getCaller()}newEntry.fireHooks()buffer = bufPool.Get()defer func() {newEntry.Buffer = nilbuffer.Reset()bufPool.Put(buffer)}()buffer.Reset()newEntry.Buffer = buffernewEntry.write()newEntry.Buffer = nil// To avoid Entry#log() returning a value that only would make sense for// panic() to use in Entry#Panic(), we avoid the allocation by checking// directly here.if level <= PanicLevel {panic(newEntry)}
}

newEntry.write() 方法就会调用 entry.Logger.Formatter.Format(entry) 方法将当前 Entry 实例格式化为字节流。注意,这里使用的 Formatter 是在创建 Logger 时指定的日志格式化器,也就是我们自定义的 CustomTextFormatter

func (entry *Entry) write() {entry.Logger.mu.Lock()defer entry.Logger.mu.Unlock()serialized, err := entry.Logger.Formatter.Format(entry)if err != nil {fmt.Fprintf(os.Stderr, "Failed to obtain reader, %v\n", err)return}if _, err := entry.Logger.Out.Write(serialized); err != nil {fmt.Fprintf(os.Stderr, "Failed to write to log, %v\n", err)}
}

CustomTextFormatter.Format() 方法我们已经实现了,根据日志级别选择设置对应的自定义颜色并打印日志消息。如果ForceColorstrue,则使用定义的颜色样式来打印日志;否则,调用父类的TextFormatter.Format方法进行默认的格式化操作:

func (f *CustomTextFormatter) Format(entry *logrus.Entry) ([]byte, error) {if f.ForceColors {switch entry.Level {case logrus.InfoLevel:f.ColorInfo.Println(entry.Message)case logrus.WarnLevel:f.ColorWarning.Println(entry.Message)case logrus.ErrorLevel:f.ColorError.Println(entry.Message)case logrus.FatalLevel, logrus.PanicLevel:f.ColorCritical.Println(entry.Message)default:f.PrintColored(entry)}return nil, nil} else {return f.TextFormatter.Format(entry)}
}

当在调用 f.ColorInfo.Println() 中,会首先调用 c.Set() 方法,将当前颜色设置为控制台输出的颜色。接着使用 defer c.unset() 语句,确保在方法执行完毕后恢复控制台输出的默认颜色。这样可以避免影响后续输出的颜色。

最后,通过调用 fmt.Fprintln(Output, a...) 方法,格式化并输出参数 a... 的内容到指定的输出位置。

func (c *Color) Println(a ...interface{}) (n int, err error) {c.Set()defer c.unset()return fmt.Fprintln(Output, a...)
}

而到 Color 结构体的 Set 方法中会先通过调用 c.isNoColorSet() 方法检查是否设置了 noColor(不使用颜色)选项。如果设置了 noColor 选项,则直接返回 Color 结构体本身,不做任何操作。

如果未设置 noColor 选项,则调用 c.format() 方法生成带有转义序列的颜色格式字符串。然后使用 fmt.Fprint 方法将该颜色格式字符串输出到 Output(继承自 log 包)中。

// Set sets the SGR sequence.
func (c *Color) Set() *Color {if c.isNoColorSet() {return c}fmt.Fprint(Output, c.format())return c
}

Color 结构体的 format 方法会首先调用 c.sequence() 方法获取当前颜色序列的 ANSI 转义序列值。然后使用 fmt.Sprintf 方法将转义序列插入到格式化字符串中,生成完整的颜色格式字符串。

func (c *Color) format() string {return fmt.Sprintf("%s[%sm", escape, c.sequence())
}

其中,escape 是一个字符转义常量,表示 ANSI 转义序列的起始字符,通常为 \033c.sequence() 方法返回的是字符串类型的 ANSI 转义序列,例如 "31" 表示红色文本。最后返回的是带有转义序列的颜色格式字符串,例如 \033[31m 表示设置文本颜色为红色。

最终调用链再执行到 fmt.Fprintln(Output, a...) 方法的 return fmt.Fprintln(Output, a...) 将格式化后的数据输出到指定的 io.Writer 接口上,并返回输出的字节数和可能的错误:

func Fprintln(w io.Writer, a ...any) (n int, err error) {p := newPrinter()p.doPrintln(a)n, err = w.Write(p.buf)p.free()return
}

整个原理就是这些!!!
感谢关注!!!

这篇关于Logrus 集成 color 库实现自定义日志颜色输出字符原理的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/151253

相关文章

Python使用python-can实现合并BLF文件

《Python使用python-can实现合并BLF文件》python-can库是Python生态中专注于CAN总线通信与数据处理的强大工具,本文将使用python-can为BLF文件合并提供高效灵活... 目录一、python-can 库:CAN 数据处理的利器二、BLF 文件合并核心代码解析1. 基础合

Python使用OpenCV实现获取视频时长的小工具

《Python使用OpenCV实现获取视频时长的小工具》在处理视频数据时,获取视频的时长是一项常见且基础的需求,本文将详细介绍如何使用Python和OpenCV获取视频时长,并对每一行代码进行深入解析... 目录一、代码实现二、代码解析1. 导入 OpenCV 库2. 定义获取视频时长的函数3. 打开视频文

golang版本升级如何实现

《golang版本升级如何实现》:本文主要介绍golang版本升级如何实现问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录golanwww.chinasem.cng版本升级linux上golang版本升级删除golang旧版本安装golang最新版本总结gola

SpringBoot中SM2公钥加密、私钥解密的实现示例详解

《SpringBoot中SM2公钥加密、私钥解密的实现示例详解》本文介绍了如何在SpringBoot项目中实现SM2公钥加密和私钥解密的功能,通过使用Hutool库和BouncyCastle依赖,简化... 目录一、前言1、加密信息(示例)2、加密结果(示例)二、实现代码1、yml文件配置2、创建SM2工具

Mysql实现范围分区表(新增、删除、重组、查看)

《Mysql实现范围分区表(新增、删除、重组、查看)》MySQL分区表的四种类型(范围、哈希、列表、键值),主要介绍了范围分区的创建、查询、添加、删除及重组织操作,具有一定的参考价值,感兴趣的可以了解... 目录一、mysql分区表分类二、范围分区(Range Partitioning1、新建分区表:2、分

MySQL 定时新增分区的实现示例

《MySQL定时新增分区的实现示例》本文主要介绍了通过存储过程和定时任务实现MySQL分区的自动创建,解决大数据量下手动维护的繁琐问题,具有一定的参考价值,感兴趣的可以了解一下... mysql创建好分区之后,有时候会需要自动创建分区。比如,一些表数据量非常大,有些数据是热点数据,按照日期分区MululbU

MySQL中查找重复值的实现

《MySQL中查找重复值的实现》查找重复值是一项常见需求,比如在数据清理、数据分析、数据质量检查等场景下,我们常常需要找出表中某列或多列的重复值,具有一定的参考价值,感兴趣的可以了解一下... 目录技术背景实现步骤方法一:使用GROUP BY和HAVING子句方法二:仅返回重复值方法三:返回完整记录方法四:

IDEA中新建/切换Git分支的实现步骤

《IDEA中新建/切换Git分支的实现步骤》本文主要介绍了IDEA中新建/切换Git分支的实现步骤,通过菜单创建新分支并选择是否切换,创建后在Git详情或右键Checkout中切换分支,感兴趣的可以了... 前提:项目已被Git托管1、点击上方栏Git->NewBrancjsh...2、输入新的分支的

怎样通过分析GC日志来定位Java进程的内存问题

《怎样通过分析GC日志来定位Java进程的内存问题》:本文主要介绍怎样通过分析GC日志来定位Java进程的内存问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、GC 日志基础配置1. 启用详细 GC 日志2. 不同收集器的日志格式二、关键指标与分析维度1.

Python实现对阿里云OSS对象存储的操作详解

《Python实现对阿里云OSS对象存储的操作详解》这篇文章主要为大家详细介绍了Python实现对阿里云OSS对象存储的操作相关知识,包括连接,上传,下载,列举等功能,感兴趣的小伙伴可以了解下... 目录一、直接使用代码二、详细使用1. 环境准备2. 初始化配置3. bucket配置创建4. 文件上传到os