mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-27 05:38:55 +00:00
17 lines
397 B
Go
17 lines
397 B
Go
package adapter
|
|
|
|
import (
|
|
"github.com/prysmaticlabs/prysm/shared/p2p"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// TracingAdapter adds a trace ID to an incoming message.
|
|
var TracingAdapter p2p.Adapter = func(next p2p.Handler) p2p.Handler {
|
|
return func(msg p2p.Message) {
|
|
var messageSpan *trace.Span
|
|
msg.Ctx, messageSpan = trace.StartSpan(msg.Ctx, "handleP2pMessage")
|
|
next(msg)
|
|
messageSpan.End()
|
|
}
|
|
}
|