Files
ziti/controller/handler_ctrl/trace.go
T
2022-02-28 15:39:27 -05:00

32 lines
840 B
Go

package handler_ctrl
import (
"github.com/golang/protobuf/proto"
"github.com/michaelquigley/pfxlog"
"github.com/openziti/channel"
"github.com/openziti/channel/trace/pb"
"github.com/openziti/fabric/pb/ctrl_pb"
"github.com/openziti/fabric/trace"
)
type traceHandler struct {
dispatcher trace.EventHandler
}
func newTraceHandler(dispatcher trace.EventHandler) *traceHandler {
return &traceHandler{dispatcher: dispatcher}
}
func (*traceHandler) ContentType() int32 {
return int32(ctrl_pb.ContentType_TraceEventType)
}
func (handler *traceHandler) HandleReceive(msg *channel.Message, _ channel.Channel) {
event := &trace_pb.ChannelMessage{}
if err := proto.Unmarshal(msg.Body, event); err == nil {
go handler.dispatcher.Accept(event)
} else {
pfxlog.Logger().Errorf("unexpected error decoding trace message (%s)", err)
}
}