mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-06 22:23:56 -05:00
5d6a406829
* Update to Go 1.23 * Update bazel version * Update rules_go * Use toolchains_protoc * Update go_honnef_go_tools * Update golang.org/x/tools * Fix violations of SA3000 * Update errcheck by re-exporting the upstream repo * Remove problematic ginkgo and gomega test helpers. Rewrote tests without these test libraries. * Update go to 1.23.5 * gofmt with go1.23.5 * Revert Patch * Unclog * Update for go 1.23 support * Fix Lint Issues * Gazelle * Fix Build * Fix Lint * no lint * Fix lint * Fix lint * Disable intrange * Preston's review --------- Co-authored-by: Preston Van Loon <preston@pvl.dev>
82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
// LogRequests logs the gRPC backend as well as request duration when the log level is set to debug
|
|
// or higher.
|
|
func LogRequests(
|
|
ctx context.Context,
|
|
method string, req,
|
|
reply interface{},
|
|
cc *grpc.ClientConn,
|
|
invoker grpc.UnaryInvoker,
|
|
opts ...grpc.CallOption,
|
|
) error {
|
|
// Shortcut when debug logging is not enabled.
|
|
if logrus.GetLevel() < logrus.DebugLevel {
|
|
return invoker(ctx, method, req, reply, cc, opts...)
|
|
}
|
|
|
|
var header metadata.MD
|
|
opts = append(
|
|
opts,
|
|
grpc.Header(&header),
|
|
)
|
|
start := time.Now()
|
|
err := invoker(ctx, method, req, reply, cc, opts...)
|
|
logrus.WithField("backend", header["x-backend"]).
|
|
WithField("method", method).WithField("duration", time.Since(start)).
|
|
Debug("gRPC request finished.")
|
|
return err
|
|
}
|
|
|
|
// LogStream prints the method at DEBUG level at the start of the stream.
|
|
func LogStream(
|
|
ctx context.Context,
|
|
sd *grpc.StreamDesc,
|
|
conn *grpc.ClientConn,
|
|
method string,
|
|
streamer grpc.Streamer,
|
|
opts ...grpc.CallOption,
|
|
) (grpc.ClientStream, error) {
|
|
// Shortcut when debug logging is not enabled.
|
|
if logrus.GetLevel() < logrus.DebugLevel {
|
|
return streamer(ctx, sd, conn, method, opts...)
|
|
}
|
|
|
|
var header metadata.MD
|
|
opts = append(
|
|
opts,
|
|
grpc.Header(&header),
|
|
)
|
|
strm, err := streamer(ctx, sd, conn, method, opts...)
|
|
logrus.WithField("backend", header["x-backend"]).
|
|
WithField("method", method).
|
|
Debug("gRPC stream started.")
|
|
return strm, err
|
|
}
|
|
|
|
// AppendHeaders parses the provided GRPC headers
|
|
// and attaches them to the provided context.
|
|
func AppendHeaders(parent context.Context, headers []string) context.Context {
|
|
for _, h := range headers {
|
|
if h != "" {
|
|
keyValue := strings.Split(h, "=")
|
|
if len(keyValue) < 2 {
|
|
logrus.Warnf("Incorrect gRPC header flag format. Skipping %v", keyValue[0])
|
|
continue
|
|
}
|
|
parent = metadata.AppendToOutgoingContext(parent, keyValue[0], strings.Join(keyValue[1:], "=")) // nolint:fatcontext
|
|
}
|
|
}
|
|
return parent
|
|
}
|