constellation/internal/logger/grpclogger.go
Daniel Weiße b25425a9f3
deps: update grpc-middleware to v2 (#2286)
Signed-off-by: Daniel Weiße <dw@edgeless.systems>
2023-08-29 14:07:19 +02:00

82 lines
1.9 KiB
Go

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package logger
import (
"fmt"
"go.uber.org/zap"
"google.golang.org/grpc/grpclog"
)
func replaceGRPCLogger(log *zap.Logger) {
gl := &grpcLogger{
logger: log.With(zap.String("system", "grpc"), zap.Bool("grpc_log", true)).WithOptions(zap.AddCallerSkip(2)),
verbosity: 0,
}
grpclog.SetLoggerV2(gl)
}
type grpcLogger struct {
logger *zap.Logger
verbosity int
}
func (l *grpcLogger) Info(args ...interface{}) {
l.logger.Info(fmt.Sprint(args...))
}
func (l *grpcLogger) Infoln(args ...interface{}) {
l.logger.Info(fmt.Sprint(args...))
}
func (l *grpcLogger) Infof(format string, args ...interface{}) {
l.logger.Info(fmt.Sprintf(format, args...))
}
func (l *grpcLogger) Warning(args ...interface{}) {
l.logger.Warn(fmt.Sprint(args...))
}
func (l *grpcLogger) Warningln(args ...interface{}) {
l.logger.Warn(fmt.Sprint(args...))
}
func (l *grpcLogger) Warningf(format string, args ...interface{}) {
l.logger.Warn(fmt.Sprintf(format, args...))
}
func (l *grpcLogger) Error(args ...interface{}) {
l.logger.Error(fmt.Sprint(args...))
}
func (l *grpcLogger) Errorln(args ...interface{}) {
l.logger.Error(fmt.Sprint(args...))
}
func (l *grpcLogger) Errorf(format string, args ...interface{}) {
l.logger.Error(fmt.Sprintf(format, args...))
}
func (l *grpcLogger) Fatal(args ...interface{}) {
l.logger.Fatal(fmt.Sprint(args...))
}
func (l *grpcLogger) Fatalln(args ...interface{}) {
l.logger.Fatal(fmt.Sprint(args...))
}
func (l *grpcLogger) Fatalf(format string, args ...interface{}) {
l.logger.Fatal(fmt.Sprintf(format, args...))
}
func (l *grpcLogger) V(level int) bool {
// Check whether the verbosity of the current log ('level') is within the specified threshold ('l.verbosity').
// As in https://github.com/grpc/grpc-go/blob/41e044e1c82fcf6a5801d6cbd7ecf952505eecb1/grpclog/loggerv2.go#L199-L201.
return level <= l.verbosity
}