7d67a90c9e
Signed-off-by: Drew Erny <drew.erny@docker.com> Upstream-commit: 1222a7081ac9ebb0830a6c8008142258c49800b5 Component: engine
32 lines
874 B
Go
32 lines
874 B
Go
package log
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"google.golang.org/grpc/grpclog"
|
|
)
|
|
|
|
type logrusWrapper struct {
|
|
*logrus.Entry
|
|
}
|
|
|
|
// V provides the functionality that returns whether a particular log level is at
|
|
// least l - this is needed to meet the LoggerV2 interface. GRPC's logging levels
|
|
// are: https://github.com/grpc/grpc-go/blob/master/grpclog/loggerv2.go#L71
|
|
// 0=info, 1=warning, 2=error, 3=fatal
|
|
// logrus's are: https://github.com/sirupsen/logrus/blob/master/logrus.go
|
|
// 0=panic, 1=fatal, 2=error, 3=warn, 4=info, 5=debug
|
|
func (lw logrusWrapper) V(l int) bool {
|
|
// translate to logrus level
|
|
logrusLevel := 4 - l
|
|
return int(lw.Logger.Level) <= logrusLevel
|
|
}
|
|
|
|
func init() {
|
|
ctx := WithModule(context.Background(), "grpc")
|
|
|
|
// completely replace the grpc logger with the logrus logger.
|
|
grpclog.SetLoggerV2(logrusWrapper{Entry: G(ctx)})
|
|
}
|