You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.1 KiB
43 lines
1.1 KiB
package tracing
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"go.opentelemetry.io/otel/trace"
|
|
"google.golang.org/grpc/peer"
|
|
"google.golang.org/grpc/stats"
|
|
)
|
|
|
|
// ClientHandler is tracing ClientHandler
|
|
type ClientHandler struct{}
|
|
|
|
// HandleConn exists to satisfy gRPC stats.Handler.
|
|
func (c *ClientHandler) HandleConn(_ context.Context, _ stats.ConnStats) {
|
|
fmt.Println("Handle connection.")
|
|
}
|
|
|
|
// TagConn exists to satisfy gRPC stats.Handler.
|
|
func (c *ClientHandler) TagConn(ctx context.Context, _ *stats.ConnTagInfo) context.Context {
|
|
return ctx
|
|
}
|
|
|
|
// HandleRPC implements per-RPC tracing and stats instrumentation.
|
|
func (c *ClientHandler) HandleRPC(ctx context.Context, rs stats.RPCStats) {
|
|
if _, ok := rs.(*stats.OutHeader); !ok {
|
|
return
|
|
}
|
|
p, ok := peer.FromContext(ctx)
|
|
if !ok {
|
|
return
|
|
}
|
|
remoteAddr := p.Addr.String()
|
|
if span := trace.SpanFromContext(ctx); span.SpanContext().IsValid() {
|
|
span.SetAttributes(peerAttr(remoteAddr)...)
|
|
}
|
|
}
|
|
|
|
// TagRPC implements per-RPC context management.
|
|
func (c *ClientHandler) TagRPC(ctx context.Context, _ *stats.RPCTagInfo) context.Context {
|
|
return ctx
|
|
}
|
|
|