mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-27 21:57:16 +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()
|
||
|
|
}
|
||
|
|
}
|