grpc/logger/context.go

45 lines
665 B
Go

package logger
import (
"context"
"sync"
"github.com/sirupsen/logrus"
)
var (
defaultLogger Logger
mu sync.RWMutex
)
type log struct{}
func init() {
defaultLogger = &logger{fl: logrus.New()}
}
func SetDefault(logger Logger) {
mu.Lock()
defer mu.Unlock()
defaultLogger = logger
}
func Set(ctx context.Context, logger Logger) context.Context {
return context.WithValue(ctx, log{}, logger)
}
func From(ctx context.Context) Logger {
log, ok := ctx.Value(log{}).(Logger)
if ok {
return log
}
if defaultLogger != nil {
return defaultLogger
}
logr := New()
mu.Lock()
defer mu.Unlock()
defaultLogger = logr
return defaultLogger
}