mirror of
https://github.com/openziti/ziti.git
synced 2026-09-10 16:55:41 +00:00
Paul Lorenz
102 lines
4.2 KiB
Go
102 lines
4.2 KiB
Go
/*
|
|
Copyright NetFoundry, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package handler_ctrl
|
|
|
|
import (
|
|
"github.com/openziti/channel"
|
|
"github.com/openziti/channel/latency"
|
|
"github.com/openziti/fabric/controller/network"
|
|
"github.com/openziti/fabric/controller/xctrl"
|
|
metrics2 "github.com/openziti/fabric/router/metrics"
|
|
"github.com/openziti/fabric/trace"
|
|
"github.com/openziti/foundation/metrics"
|
|
"time"
|
|
)
|
|
|
|
type bindHandler struct {
|
|
router *network.Router
|
|
network *network.Network
|
|
xctrls []xctrl.Xctrl
|
|
}
|
|
|
|
func newBindHandler(router *network.Router, network *network.Network, xctrls []xctrl.Xctrl) channel.BindHandler {
|
|
return &bindHandler{router: router, network: network, xctrls: xctrls}
|
|
}
|
|
|
|
func (self *bindHandler) BindChannel(binding channel.Binding) error {
|
|
traceDispatchWrapper := trace.NewDispatchWrapper(self.network.GetEventDispatcher().Dispatch)
|
|
binding.AddTypedReceiveHandler(newCircuitRequestHandler(self.router, self.network))
|
|
binding.AddTypedReceiveHandler(newRouteResultHandler(self.network, self.router))
|
|
binding.AddTypedReceiveHandler(newCircuitConfirmationHandler(self.network, self.router))
|
|
binding.AddTypedReceiveHandler(newCreateTerminatorHandler(self.network, self.router))
|
|
binding.AddTypedReceiveHandler(newRemoveTerminatorHandler(self.network))
|
|
binding.AddTypedReceiveHandler(newUpdateTerminatorHandler(self.network))
|
|
binding.AddTypedReceiveHandler(newLinkConnectedHandler(self.router, self.network))
|
|
binding.AddTypedReceiveHandler(newRouterLinkHandler(self.router, self.network))
|
|
binding.AddTypedReceiveHandler(newVerifyLinkHandler(self.router, self.network))
|
|
binding.AddTypedReceiveHandler(newFaultHandler(self.router, self.network))
|
|
binding.AddTypedReceiveHandler(newMetricsHandler(self.network))
|
|
binding.AddTypedReceiveHandler(newTraceHandler(traceDispatchWrapper))
|
|
binding.AddTypedReceiveHandler(newInspectHandler(self.network))
|
|
binding.AddTypedReceiveHandler(newPingHandler())
|
|
binding.AddPeekHandler(trace.NewChannelPeekHandler(self.network.GetAppId(), binding.GetChannel(), self.network.GetTraceController(), traceDispatchWrapper))
|
|
binding.AddPeekHandler(metrics2.NewCtrlChannelPeekHandler(self.router.Id, self.network.GetMetricsRegistry()))
|
|
|
|
if self.router.VersionInfo.HasMinimumVersion("0.18.7") {
|
|
roundTripHistogram := self.network.GetMetricsRegistry().Histogram("ctrl.latency:" + self.router.Id)
|
|
queueTimeHistogram := self.network.GetMetricsRegistry().Histogram("ctrl.queue_time:" + self.router.Id)
|
|
latencyHandler := &ctrlChannelLatencyHandler{
|
|
roundTripHistogram: roundTripHistogram,
|
|
queueTimeHistogram: queueTimeHistogram,
|
|
}
|
|
latency.AddLatencyProbe(binding.GetChannel(), binding, self.network.GetOptions().CtrlChanLatencyInterval/time.Duration(10), 10, latencyHandler.HandleLatency)
|
|
binding.AddCloseHandler(channel.CloseHandlerF(func(ch channel.Channel) {
|
|
roundTripHistogram.Dispose()
|
|
queueTimeHistogram.Dispose()
|
|
}))
|
|
}
|
|
|
|
xctrlDone := make(chan struct{})
|
|
for _, x := range self.xctrls {
|
|
if err := binding.Bind(x); err != nil {
|
|
return err
|
|
}
|
|
if err := x.Run(binding.GetChannel(), self.network.GetDb(), xctrlDone); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if len(self.xctrls) > 0 {
|
|
binding.AddCloseHandler(newXctrlCloseHandler(xctrlDone))
|
|
}
|
|
|
|
binding.AddCloseHandler(newCloseHandler(self.router, self.network))
|
|
return nil
|
|
}
|
|
|
|
type ctrlChannelLatencyHandler struct {
|
|
roundTripHistogram metrics.Histogram
|
|
queueTimeHistogram metrics.Histogram
|
|
}
|
|
|
|
func (self *ctrlChannelLatencyHandler) HandleLatency(latencyType latency.Type, elapsed time.Duration) {
|
|
if latencyType == latency.RoundTripType {
|
|
self.roundTripHistogram.Update(elapsed.Nanoseconds())
|
|
} else if latencyType == latency.BeforeSendType {
|
|
self.queueTimeHistogram.Update(elapsed.Nanoseconds())
|
|
}
|
|
}
|