go-zero/zrpc/internal/rpclogger.go

package internal

import (
	"sync"

	"github.com/tal-tech/go-zero/core/logx"
	"google.golang.org/grpc/grpclog"
)

// because grpclog.errorLog is not exported, we need to define our own.
const errorLevel = 2

var once sync.Once

// A Logger is a rpc logger.
type Logger struct{}

// InitLogger initializes the rpc logger.
func InitLogger() {
	once.Do(func() {
		grpclog.SetLoggerV2(new(Logger))
	})
}

// Error logs the given args into error log.
func (l *Logger) Error(args ...interface{}) {
	logx.Error(args...)
}

// Errorf logs the given args with format into error log.
func (l *Logger) Errorf(format string, args ...interface{}) {
	logx.Errorf(format, args...)
}

// Errorln logs the given args into error log with newline.
func (l *Logger) Errorln(args ...interface{}) {
	logx.Error(args...)
}

// Fatal logs the given args into error log.
func (l *Logger) Fatal(args ...interface{}) {
	logx.Error(args...)
}

// Fatalf logs the given args with format into error log.
func (l *Logger) Fatalf(format string, args ...interface{}) {
	logx.Errorf(format, args...)
}

// Fatalln logs args into error log with newline.
func (l *Logger) Fatalln(args ...interface{}) {
	logx.Error(args...)
}

// Info ignores the grpc info logs.
func (l *Logger) Info(args ...interface{}) {
	// ignore builtin grpc info
}

// Infoln ignores the grpc info logs.
func (l *Logger) Infoln(args ...interface{}) {
	// ignore builtin grpc info
}

// Infof ignores the grpc info logs.
func (l *Logger) Infof(format string, args ...interface{}) {
	// ignore builtin grpc info
}

// V checks if meet required log level.
func (l *Logger) V(v int) bool {
	return v >= errorLevel
}

// Warning ignores the grpc warning logs.
func (l *Logger) Warning(args ...interface{}) {
	// ignore builtin grpc warning
}

// Warningf ignores the grpc warning logs.
func (l *Logger) Warningf(format string, args ...interface{}) {
	// ignore builtin grpc warning
}

// Warningln ignores the grpc warning logs.
func (l *Logger) Warningln(args ...interface{}) {
	// ignore builtin grpc warning
}