mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
@@ -2,27 +2,34 @@ package interceptors
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
"github.com/grafana/grafana/pkg/infra/log"
|
"github.com/grafana/grafana/pkg/infra/log"
|
||||||
|
"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/logging"
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
)
|
)
|
||||||
|
|
||||||
func LoggingUnaryInterceptor(logger log.Logger, enabled bool) grpc.UnaryServerInterceptor {
|
func InterceptorLogger(l log.Logger, enabled bool) logging.Logger {
|
||||||
return func(
|
return logging.LoggerFunc(func(ctx context.Context, lvl logging.Level, msg string, fields ...any) {
|
||||||
ctx context.Context,
|
if !enabled {
|
||||||
req any,
|
return
|
||||||
info *grpc.UnaryServerInfo,
|
|
||||||
handler grpc.UnaryHandler,
|
|
||||||
) (resp any, err error) {
|
|
||||||
resp, err = handler(ctx, req)
|
|
||||||
if enabled {
|
|
||||||
ctxLogger := logger.FromContext(ctx)
|
|
||||||
if err != nil {
|
|
||||||
ctxLogger.Error("gRPC call", "method", info.FullMethod, "req", req, "err", err)
|
|
||||||
} else {
|
|
||||||
ctxLogger.Info("gRPC call", "method", info.FullMethod, "req", req, "resp", resp)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return resp, err
|
l := l.FromContext(ctx)
|
||||||
}
|
switch lvl {
|
||||||
|
case logging.LevelDebug:
|
||||||
|
l.Debug(msg, fields...)
|
||||||
|
case logging.LevelInfo:
|
||||||
|
l.Info(msg, fields...)
|
||||||
|
case logging.LevelWarn:
|
||||||
|
l.Warn(msg, fields...)
|
||||||
|
case logging.LevelError:
|
||||||
|
l.Error(msg, fields...)
|
||||||
|
default:
|
||||||
|
panic(fmt.Sprintf("unknown level %v", lvl))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoggingUnaryInterceptor(logger log.Logger, enabled bool) grpc.UnaryServerInterceptor {
|
||||||
|
return logging.UnaryServerInterceptor(InterceptorLogger(logger, enabled))
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user