netbird/client/grpc/logger/logger.go

package logger

import (
	"context"
	"os"
	"time"

	log "github.com/sirupsen/logrus"
	"google.golang.org/grpc"
	"google.golang.org/protobuf/encoding/protojson"
	"google.golang.org/protobuf/proto"
)

// UnaryClientInterceptor logs gRPC requests using the global logrus logger
func UnaryClientInterceptor() grpc.UnaryClientInterceptor {
	return func(
		ctx context.Context,
		method string,
		req, reply interface{},
		cc *grpc.ClientConn,
		invoker grpc.UnaryInvoker,
		opts ...grpc.CallOption,
	) error {
		start := time.Now()

		// don't log if log output is not a file
		ignoreLog := log.StandardLogger().Out == os.Stdout || log.StandardLogger().Out == os.Stderr

		if !ignoreLog {
			// log the request
			if msg, ok := req.(proto.Message); ok {
				if jsonReq, err := protojson.Marshal(msg); err == nil {
					log.Debugf("gRPC request initiated: method=%s, request=%s", method, jsonReq)
				} else {
					log.Warnf("Could not marshal gRPC request: method=%s, error=%v", method, err)
				}
			} else {
				log.Debugf("gRPC request initiated: method=%s, requestType=%T", method, req)
			}
		}

		err := invoker(ctx, method, req, reply, cc, opts...)

		duration := time.Since(start)

		// log the response
		if !ignoreLog {
			if err != nil {
				log.Errorf("gRPC request failed: method=%s, duration=%v, error=%v", method, duration, err)
			} else {
				if msg, ok := reply.(proto.Message); ok {
					if jsonReply, err := protojson.Marshal(msg); err == nil {
						log.Debugf("gRPC request succeeded: method=%s, duration=%v, response=%s", method, duration, jsonReply)
					} else {
						log.Warnf("Could not marshal gRPC response: method=%s, error=%v", method, err)
					}
				} else {
					log.Debugf("gRPC request succeeded: method=%s, duration=%v, responseType=%T", method, duration, reply)
				}
			}
		}

		return err
	}
}