在go-grpc-middleware中实现选择性日志记录的方法

在gRPC服务开发中，日志记录是一个非常重要的功能，它帮助我们监控服务运行状态和排查问题。然而，在某些场景下，我们可能希望对某些特定的方法调用不记录日志或改变其日志级别，例如频繁调用的健康检查接口。

go-grpc-middleware项目提供了强大的gRPC中间件功能，其中日志记录中间件在v2版本中进行了重构。与v1版本不同，v2版本移除了基于方法名决定是否记录日志的"decider"函数，转而推荐使用selector拦截器来实现更灵活的控制。

为什么需要选择性日志记录

健康检查接口通常会被频繁调用（可能每秒多次），如果每次都记录完整日志，会导致：

1. 日志系统产生大量冗余数据
2. 增加存储成本
3. 降低日志系统的处理效率
4. 可能掩盖真正重要的日志信息

实现方案

在go-grpc-middleware v2中，我们可以通过组合使用日志拦截器和selector拦截器来实现选择性日志记录：

import (
    "context"
    
    "github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/logging"
    "github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/selector"
)

// 定义健康检查方法名
const healthCheckMethod = "/grpc.health.v1.Health/Check"

// 创建日志拦截器
logOpts := []logging.Option{
    logging.WithLogOnEvents(logging.StartCall, logging.FinishCall),
}

// 创建selector匹配函数
matchFunc := func(ctx context.Context, callMeta logging.CallMeta) bool {
    return callMeta.FullMethod != healthCheckMethod
}

// 组合中间件
loggingInterceptor := selector.MatchFunc(matchFunc, logging.UnaryServerInterceptor(logOpts...))

进阶用法

除了完全跳过日志记录外，我们还可以为不同方法设置不同的日志级别：

matchFunc := func(ctx context.Context, callMeta logging.CallMeta) bool {
    if callMeta.FullMethod == healthCheckMethod {
        // 为健康检查设置更高级别的日志
        return logging.LevelError
    }
    // 其他方法保持默认级别
    return logging.LevelDebug
}

最佳实践建议

1. 对于高频调用的方法（如健康检查、心跳等），建议完全跳过日志记录或设置为更高级别
2. 对于关键业务方法，建议记录更详细的日志
3. 可以考虑基于请求频率动态调整日志级别
4. 在生产环境中，应该通过配置文件而不是硬编码来控制日志行为

通过这种灵活的选择性日志记录机制，我们可以在保证必要监控能力的同时，有效控制日志系统的负载和存储成本。