文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

如何使用GO语言从零实现日志包

2024-11-30 16:00

关注

背景

当前的开源日志包有很多,像go中的标准库log包、glog、logrus、zap。它们每种日志包都有相应的应用场景。四种日志包相关对比如下所示:

标准库log

功能简单,不支持日志级别、日志格式。但是使用简单,易于快速上手。大型项目较少使用

glog

提供了日志包的基本功能,像日志级别、格式等。适合一些小项目

logrus

功能强大,不仅实现了基本日志功能,还提供了很多高级功能。适合大型项目

zap

功能强大,性能高,适合对日志性能要求高的项目。另外zap的子包zapcore提供了很多底层日志接口,适合二次开发

从头开发一个日志包,可以让我们了解日志包的底层逻辑,使得我们对日志包有定制需求的时候,可以能够基于开源的日志包实现我们的功能。所以本文以cuslog为例,看看如何实现我们自己的日志包(代码:
https://github.com/marmotedu/gopractise-demo/tree/master/log/cuslog)。

代码结构

cuslog目录代码结构

cuslog代码结构

要实现一个日志包,就需要实现下面三个基本的对象,Entry,Logger,Options。

Entry

代码:
https://github.com/marmotedu/gopractise-demo/blob/master/log/cuslog/entry.go。

//构造函数
func entry(logger *logger) *Entry {
return &Entry{logger: logger, Buffer: new(bytes.Buffer), Map: make(map[string]interface{}, 5)}
}

func (e *Entry) write(level Level, format string, args ...interface{}) {
if e.logger.opt.level > level {
return
}
e.Time = time.Now()
e.Level = level
e.Format = format
e.Args = args
if !e.logger.opt.disableCaller {
if pc, file, line, ok := runtime.Caller(2); !ok {
e.File = "???"
e.Func = "???"
} else {
e.File, e.Line, e.Func = file, line, runtime.FuncForPC(pc).Name()
e.Func = e.Func[strings.LastIndex(e.Func, "/")+1:]
}
}
e.format()
e.writer()
e.release()
}

Entry的write方法实现了将它Buffer中的数据,写入到它的logger所配置的output中。

Logger

代码:
https://github.com/marmotedu/gopractise-demo/blob/master/log/cuslog/logger.go。


func New(opts ...Option) *logger {
logger := &logger{opt: initOptions(opts...)}
logger.entryPool = &sync.Pool{New: func() interface{} { return entry(logger) }}
return logger
}

Options

代码:
https://github.com/marmotedu/gopractise-demo/blob/68e100ee78a3093e6f2434439e7d4b143b9ebf60/log/cuslog/options.go。


type Option func(*options)

func initOptions(opts ...Option) (o *options) {
o = &options{}
for _, opt := range opts {
opt(o)
}

if o.output == nil {
o.output = os.Stderr
}

if o.formatter == nil {
o.formatter = &TextFormatter{}
}

return
}

func WithOutput(output io.Writer) Option {
return func(o *options) {
o.output = output
}
}

func WithLevel(level Level) Option {
return func(o *options) {
o.level = level
}
}

应用

通过下面的代码,我们看看整个代码是如何串起来的。

// 输出到文件
fd, err := os.OpenFile("test.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
log.Fatalln("create file test.log failed")
}
defer fd.Close()

l := cuslog.New(cuslog.WithLevel(cuslog.InfoLevel),
cuslog.WithOutput(fd),
cuslog.WithFormatter(&cuslog.JsonFormatter{IgnoreBasicFields: false}),
)
l.Info("custom log with json formatter")
  1. 上面的整体实现是,将"custom log with json formatter"这段字符串写入到指定文件里面
  2. 1到6行创建并打开文件
  3. 重点是8行,cuslog.WithLevel(cuslog.InfoLevel)、cuslog.WithOutput(fd)、cuslog.WithFormatter(&cuslog.JsonFormatter{IgnoreBasicFields: false})三个函数调用,返回三个函数func(o *options),这些函数都是Option类型,因为type Option func(*options)。
  4. 然后调用cuslog.New,这个函数上面也给出了,它里面通过initOptions依次调用上面的Option函数,对options对象进行设置,然后把options赋给opt, 并创建logger. logger := &logger{opt: initOptions(opts...)}
  5. 最后调用l.Info("custom log with json formatter")把字符串输出到文件中.整个调用链是l.info===>通过l的pool获取entry===>调用entry的write(InfoLevel, FmtEmptySeparate, args...),在这个write函数里面,只有infoLevel比logger.level优先级大或相等,才输出。并且通过runtime.Caller(2)获取最上层调用info时的,文件名,行号,函数名等信息。因为这个地方有2层嵌套才调用到entry的write,所以runtime.Caller(2)的参数是2

总结

上面的代码实现了基本的日志功能,包括日志级别、日志格式配置、输出文件或标准输出的设置。但是一些高级的功能,比如按级别分类输出,Hook能力,结构化日志。这些目前都不支持。

来源:今日头条内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯