Golang如何使用gRPC拦截器实现日志收集_Golang gRPC拦截器日志收集实践

func loggingUnaryInterceptor(
    ctx context.Context,
    req interface{},
    info *grpc.UnaryServerInfo,
    handler grpc.UnaryHandler,
) (interface{}, error) {
    // 记录请求开始
    start := time.Now()
    <strong>log.Printf("Started %s at %v", info.FullMethod, start)</strong>

    // 调用实际的处理函数
    resp, err := handler(ctx, req)

    // 记录请求结束
    duration := time.Since(start)
    if err != nil {
        <strong>log.Printf("Completed %s with error: %v, duration: %v", info.FullMethod, err, duration)</strong>
    } else {
        <strong>log.Printf("Completed %s successfully, duration: %v", info.FullMethod, duration)</strong>
    }

    return resp, err
}

server := grpc.NewServer(
    grpc.UnaryInterceptor(loggingUnaryInterceptor),
)

import "github.com/grpc-ecosystem/go-grpc-middleware"

server := grpc.NewServer(
    grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
        loggingUnaryInterceptor,
        authUnaryInterceptor,
        recoveryInterceptor,
    )),
)

if peer, ok := peer.FromContext(ctx); ok {
    log.Printf("Client address: %s", peer.Addr.String())
}