forked from asynkron/protoactor-go
/
sendermiddleware.go
31 lines (26 loc) · 1.16 KB
/
sendermiddleware.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
package opentracing
import (
"github.com/qjpcpu/protoactor-go/actor"
"github.com/qjpcpu/protoactor-go/log"
"github.com/opentracing/opentracing-go"
)
func SenderMiddleware() actor.SenderMiddleware {
return func(next actor.SenderFunc) actor.SenderFunc {
return func(c actor.SenderContext, target *actor.PID, envelope *actor.MessageEnvelope) {
span := getActiveSpan(c.Self())
if span == nil {
logger.Debug("OUTBOUND No active span", log.Stringer("PID", c.Self()), log.TypeOf("ActorType", c.Actor()), log.TypeOf("MessageType", envelope.Message))
next(c, target, envelope)
return
}
err := opentracing.GlobalTracer().Inject(span.Context(), opentracing.TextMap, opentracing.TextMapWriter(&messageEnvelopeWriter{MessageEnvelope: envelope}))
if err != nil {
logger.Debug("OUTBOUND Error injecting", log.Stringer("PID", c.Self()), log.TypeOf("ActorType", c.Actor()), log.TypeOf("MessageType", envelope.Message))
next(c, target, envelope)
return
}
logger.Debug("OUTBOUND Successfully injected", log.Stringer("PID", c.Self()), log.TypeOf("ActorType", c.Actor()), log.TypeOf("MessageType", envelope.Message))
next(c, target, envelope)
}
}
}