源码学习 go 语言的 log 包

日志的重要性

当一个程序出了问题时,不同的人会有不同反应。

有的人抓耳挠腮一脸茫然,有的人开始人肉跟踪代码试图找出破绽,还有的人已经开始准备在程序里加 print 了。

Print 是个好东西,它几乎可以让我们像 debugger 那样跟踪查看代码执行到某行时变量的值。

但是当代码跑到生产环境后,这种做法就歇菜了。生产环境要比本地复杂地多,有些问题不可重现,而且我们根本不可能跑到生产环境旁若无人的改代码。

这时候怎么办呢?使用日志。将重要的信息记录下来,调试信息、警告信息和错误信息都很有价值。

Go 语言中的 log 包

日志多被记录到日志文件。写日志文件其实没什么高深的东西,甚至在很多没有内置库支持的场合,很多人只是把写文件操作简单包装一下。

Go 语言为日志功能提供了一个简单的包。这里的“简单”并非出于谦虚,log/log.go 源代码文件只有346行。但由于日志对象使用了 interface 的设计,实现足够灵活。

在分析该文件时,我们依然按照先 struct 后 func 的顺序来一一展开。

Logger 结构体

Log 包支持同时存在多个 logger,这些 logger 可以有不同的输出格式且输出内容互不干扰。

我们来看一下 Logger 的定义:

type Logger struct {
    mu     sync.Mutex // ensures atomic writes; protects the following fields
    prefix string     // prefix to write at beginning of each line
    flag   int        // properties
    out    io.Writer  // destination for output
    buf    []byte     // for accumulating text to write
}

我们先重点关注 flag、out 和 buf 这三个字段。

flag 控制日志信息的前缀,比如是否显示时间、是否显示打印日志的文件名和行号等。flag 可取如下一些值。

const (
    Ldate         = 1 << iota
    Ltime
    Lmicroseconds
    Llongfile
    Lshortfile
    LUTC
    LstdFlags     = Ldate | Ltime
)

out 字段是一个 io.Writer,在 io/io.go 包中它是下面这个样子:

type Writer interface {
    Write(p []byte) (n int, err error)
}

out 表示我们要把日志写到哪里。它是一个 Writer,即只要实现 Write() 方法即可。这为日志的使用提供了很大的便利。

设想一下,假如我们想通过 log 模块将日志写入数据库该怎么做呢?只需要实现一个如上所示的 Write() 方法,该方法接收一条日志时,将该日志写入数据库,并返回写入字节数和错误信息即可。

buf 被用来操作临时的日志信息。

Log 包提供来一个 New() 函数来实例化一个 Logger。

func New(out io.Writer, prefix string, flag int) *Logger {
    return &Logger{out: out, prefix: prefix, flag: flag}
}

写日志方法

log 包里的写日志方法都是对 Logger.Output() 方法的简单包装。下面我们来看一下 Logger.Output() 方法。

func (l *Logger) Output(calldepth int, s string) error {
    now := time.Now() // get this early.
    var file string
    var line int

    // ...

        _, file, line, ok = runtime.Caller(calldepth)

        // ...

    l.buf = l.buf[:0]
    l.formatHeader(&l.buf, now, file, line)
    l.buf = append(l.buf, s...)
    if len(s) == 0 || s[len(s)-1] != '\n' {
        l.buf = append(l.buf, '\n')
    }
    _, err := l.out.Write(l.buf)
    return err
}

代码还是非常简单的,将调用 log 的时间、文件名、行号及日志信息写入 out 即完成了日志的输出。

其中 runtime.Caller() 用于获取调用者的信息。

其中 // ... 是我们省略的一些在此处不必要的细节。这里我们一会再来讨论。

Log 包使用 demo

下面我们针对之前分析的代码写一个演示代码:

package main

import (
    "log"
    "os"
)

func main() {
    logger := log.New(os.Stderr, "", log.Lshortfile | log.LstdFlags)

    logger.Output(1, "hello from demo")
}

我们从标准错误创建一个 logger,并调用 Logger.Output() 将日志打印到标准错误。注意,Logger.Output() 的第一个参数是 1,这样刚好找到我们的调用位置。

执行后系统输出:2017/09/11 22:31:10 demo.go:11: hello from demo。

对调用函数进行封装

直接调用 Logger.Output() 不够友好,特别是还携带了一个充满 hack 意味的 calldepth 参数。

Log 包以 Output() 为基础,提供了很多其它实用工具。实现方式简单且大致相同,下面摘几个出来看一下。

func (l *Logger) Print(v ...interface{}) { l.Output(2, fmt.Sprint(v...)) }

func (l *Logger) Fatal(v ...interface{}) {
    l.Output(2, fmt.Sprint(v...))
    os.Exit(1)
}

func (l *Logger) Panic(v ...interface{}) {
    s := fmt.Sprint(v...)
    l.Output(2, s)
    panic(s)
}

与 flag 包相同(参见《源码学习 go 语言的 flag 包》),log 包同样实现了一套顶层函数。

var std = New(os.Stderr, "", LstdFlags)

func Print(v ...interface{}) {
    std.Output(2, fmt.Sprint(v...))
}

这下简单了,上面的 demo 可以重写为如下格式。

package main

import (
    "log"
)

func main() {
    log.SetFlags(log.Lshortfile | log.LstdFlags)

    log.Print("hello from demo")
}

对于一些简单的程序,这套函数已经足够了。

如果是要写到文件,可以考虑在程序里重定向标准错误,或手动创建一个 Logger。

Next?

log/log.go 里有一个格式化函数 Logger.formatHeader(),里面处理了日志前缀的格式。代码很简单,而且功能相对独立,这里就不分析了。

我们分析了 log 包的 log.go 文件,但实际上,log 包还对 syslog 做了支持,代码在 log/syslog/ 下。有兴趣可以看一下。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,904评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,581评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,527评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,463评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,546评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,572评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,582评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,330评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,776评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,087评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,257评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,923评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,571评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,192评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,436评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,145评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,127评论 2 352

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,647评论 18 139
  • 在应用程序中添加日志记录总的来说基于三个目的:监视代码中变量的变化情况,周期性的记录到文件中供其他应用进行统计分析...
    时待吾阅读 4,982评论 0 6
  • 在应用程序中添加日志记录总的来说基于三个目的:监视代码中变量的变化情况,周期性的记录到文件中供其他应用进行统计分析...
    时待吾阅读 5,020评论 1 13
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 172,001评论 25 707
  • 《父亲节献礼》 文/伊万 老爸老爸真厉害,你永远是我最爱! 妈妈经常夸我帅,你若不帅我怎帅?! 厨房是您的挚爱,刀...
    伊万公津阅读 232评论 0 1