Question

轻量级日志记录器。

• Print
• Panic : Print + panic
• Fatal : Print + os.Exit(1)

func main() {
	l := log.New(os.Stdout, "[demo] ", log.Ldate | log.Lshortfile)
	l.Println("hello world")
}

源码剖析

默认输出的 Stderr ，可通过 SetOutput 修改。

// log.go

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

func Default() *Logger { return std }

func SetOutput(w io.Writer) {
	std.SetOutput(w)
}

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

内部实现并发安全。

type Logger struct {
	mu        sync.Mutex
	prefix    string
	flag      int
	out       io.Writer
	buf       []byte
	isDiscard int32
}

func (l *Logger) Print(v ...any) {
	if atomic.LoadInt32(&l.isDiscard) != 0 { return }
	l.Output(2, fmt.Sprint(v...))
}

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

func (l *Logger) Output(calldepth int, s string) error {
	l.mu.Lock()
	defer l.mu.Unlock()
    
	if l.flag & (Lshortfile|Llongfile) != 0 {
		l.mu.Unlock()
        
        // 忽略 Print/Fatal + Output，获取用户调用帧。
		_, file, line, ok = runtime.Caller(calldepth)
        
		l.mu.Lock()
	}
    
	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
}