gRPC Zap ReplaceGrpcLoggerV2

logging/zap/doc.go

@@ -15,7 +15,7 @@ logged as structured `jsonpb` fields for every message received/sent (both unary
 `Payload*Interceptor` functions for that. Please note that the user-provided function that determines whetether to log
 the full request/response payload needs to be written with care, this can significantly slow down gRPC.
 
-ZAP can also be made as a backend for gRPC library internals. For that use `ReplaceGrpcLogger`.
+ZAP can also be made as a backend for gRPC library internals. For that use `ReplaceGrpcLoggerV2`.
 
 
 *Server Interceptor*

logging/zap/examples_test.go

@@ -4,10 +4,10 @@ import (
  "context"
  "time"
 
- "github.com/grpc-ecosystem/go-grpc-middleware"
- "github.com/grpc-ecosystem/go-grpc-middleware/logging/zap"
- "github.com/grpc-ecosystem/go-grpc-middleware/tags"
- "github.com/grpc-ecosystem/go-grpc-middleware/tags/zap"
+ grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
+ grpc_zap "github.com/grpc-ecosystem/go-grpc-middleware/logging/zap"
+ grpc_ctxtags "github.com/grpc-ecosystem/go-grpc-middleware/tags"
+ ctx_zap "github.com/grpc-ecosystem/go-grpc-middleware/tags/zap"
  pb_testproto "github.com/grpc-ecosystem/go-grpc-middleware/testing/testproto"
  "go.uber.org/zap"
  "go.uber.org/zap/zapcore"
@@ -26,7 +26,7 @@ func Example_initialization() {
  grpc_zap.WithLevels(customFunc),
  }
  // Make sure that log statements internal to gRPC library are logged using the zapLogger as well.
- grpc_zap.ReplaceGrpcLogger(zapLogger)
+ grpc_zap.ReplaceGrpcLoggerV2(zapLogger, grpc_zap.DefaultGrpcLoggerV2Options)
  // Create a server, make sure we put the grpc_ctxtags context before everything else.
  _ = grpc.NewServer(
  grpc_middleware.WithUnaryServerChain(

logging/zap/grpclogger.go

@@ -12,6 +12,7 @@ import (
 
 // ReplaceGrpcLogger sets the given zap.Logger as a gRPC-level logger.
 // This should be called *before* any other initialization, preferably from init() functions.
+// Deprecated: use ReplaceGrpcLoggerV2.
 func ReplaceGrpcLogger(logger *zap.Logger) {
  zgl := &zapGrpcLogger{logger.With(SystemField, zap.Bool("grpc_log", true))}
  grpclog.SetLogger(zgl)
@@ -44,3 +45,80 @@ func (l *zapGrpcLogger) Printf(format string, args ...interface{}) {
 func (l *zapGrpcLogger) Println(args ...interface{}) {
  l.logger.Info(fmt.Sprint(args...))
 }
+
+// GrpcLoggerV2Options allows you to set grpclogger options when using ReplaceGrpcLoggerV2.
+type GrpcLoggerV2Options struct {
+ verbosity int
+}
+
+// DefaultGrpcLoggerV2Options is a set of default options for ReplaceGrpcLoggerV2.
+var DefaultGrpcLoggerV2Options = &GrpcLoggerV2Options{}
+
+// ReplaceGrpcLoggerV2 replaces the grpc_log.LoggerV2 with the input zap.Logger.
+// This logger allows you to set grpc verbosity level.
+func ReplaceGrpcLoggerV2(logger *zap.Logger, opts *GrpcLoggerV2Options) {
+ zgl := &zapGrpcLoggerV2{Logger: logger.With(SystemField, zap.Bool("grpc_log", true))}
+
+ if opts != nil {
+ zgl.verbosity = opts.verbosity
+ }
+
+ grpclog.SetLoggerV2(zgl)
+}
+
+type zapGrpcLoggerV2 struct {
+ *zap.Logger
+ verbosity int
+}
+
+func (l *zapGrpcLoggerV2) Info(args ...interface{}) {
+ l.Logger.Info(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Infoln(args ...interface{}) {
+ l.Logger.Info(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Infof(format string, args ...interface{}) {
+ l.Logger.Info(fmt.Sprintf(format, args...))
+}
+
+func (l *zapGrpcLoggerV2) Warning(args ...interface{}) {
+ l.Logger.Warn(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Warningln(args ...interface{}) {
+ l.Logger.Warn(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Warningf(format string, args ...interface{}) {
+ l.Logger.Warn(fmt.Sprintf(format, args...))
+}
+
+func (l *zapGrpcLoggerV2) Error(args ...interface{}) {
+ l.Logger.Error(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Errorln(args ...interface{}) {
+ l.Logger.Error(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Errorf(format string, args ...interface{}) {
+ l.Logger.Error(fmt.Sprintf(format, args...))
+}
+
+func (l *zapGrpcLoggerV2) Fatal(args ...interface{}) {
+ l.Logger.Fatal(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Fatalln(args ...interface{}) {
+ l.Logger.Fatal(fmt.Sprint(args...))
+}
+
+func (l *zapGrpcLoggerV2) Fatalf(format string, args ...interface{}) {
+ l.Logger.Fatal(fmt.Sprintf(format, args...))
+}
+
+func (l *zapGrpcLoggerV2) V(level int) bool {
+ return level <= l.verbosity
+}