mirror of
https://github.com/linka-cloud/grpc.git
synced 2025-06-22 09:12:28 +00:00
service and client implements last grpc server and client interfaces, improved errors, add signals handling
This commit is contained in:
45
logger/context.go
Normal file
45
logger/context.go
Normal file
@ -0,0 +1,45 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
var (
|
||||
defaultLogger Logger
|
||||
mu sync.RWMutex
|
||||
)
|
||||
|
||||
type log struct{}
|
||||
|
||||
func init() {
|
||||
defaultLogger = &logger{FieldLogger: 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
|
||||
}
|
||||
|
Reference in New Issue
Block a user