log/context.go

package log

import (
	"context"
)

type key int

const (
	logContextKey key = iota
)

// WithContext 把命令行输出logger保存到context中.可以使用FromContext(ctx)获取绑定的logger
func WithContext(ctx context.Context) context.Context {
	return std.WithContext(ctx)
}

// WithContext 把指定的logger保存到context中.可以使用FromContext(ctx)获取绑定的logger
func (l *zapLogger) WithContext(ctx context.Context) context.Context {
	return context.WithValue(ctx, logContextKey, l)
}

// FromContext 从ctx中获取logger
func FromContext(ctx context.Context) Logger {
	if ctx != nil {
		logger := ctx.Value(logContextKey)
		if logger != nil {
			return logger.(Logger)
		}
	}

	return WithName("Unknown-Context")
}