mirror of
https://github.com/openziti/ziti.git
synced 2026-09-10 16:55:41 +00:00
Paul Lorenz
1c122af490
- moves the channel dependency to channel/v5 v5.0.10 and sdk-golang to v1.9.0 in the root and zititest modules - mechanically rewrites every channel/v4 import path to channel/v5 This is the import-path-only step; the API-level changes the switch requires land in the following commit. This commit does not build on its own.
97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/michaelquigley/pfxlog"
|
|
"github.com/openziti/channel/v5"
|
|
"github.com/openziti/channel/v5/protobufs"
|
|
"github.com/openziti/ziti/v2/common/pb/ctrl_pb"
|
|
config2 "github.com/openziti/ziti/v2/controller/config"
|
|
"github.com/openziti/ziti/v2/controller/model"
|
|
"github.com/openziti/ziti/v2/controller/raft"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// OnConnectSettingsHandler sends a ctrl_pb.ContentType_SettingsType message when routers connect if necessary
|
|
// Settings are a map of int32 -> []byte data. The type should be used to determine how the setting's []byte
|
|
// array is consumed.
|
|
type OnConnectSettingsHandler struct {
|
|
config *config2.Config
|
|
settings map[int32][]byte
|
|
}
|
|
|
|
func (o *OnConnectSettingsHandler) RouterDisconnected(r *model.Router) {
|
|
//do nothing, satisfy interface
|
|
}
|
|
|
|
func (o OnConnectSettingsHandler) RouterConnected(r *model.Router) {
|
|
if len(o.settings) > 0 {
|
|
settingsMsg := &ctrl_pb.Settings{
|
|
Data: map[int32][]byte{},
|
|
}
|
|
|
|
for k, v := range o.settings {
|
|
settingsMsg.Data[k] = v
|
|
}
|
|
|
|
if body, err := proto.Marshal(settingsMsg); err == nil {
|
|
msg := channel.NewMessage(int32(ctrl_pb.ContentType_SettingsType), body)
|
|
if err := r.Control.GetDefaultSender().Send(msg); err == nil {
|
|
pfxlog.Logger().WithError(err).WithFields(map[string]interface{}{
|
|
"routerId": r.Id,
|
|
"channel": r.Control.GetChannel().LogicalName(),
|
|
}).Error("error sending settings on router connect")
|
|
}
|
|
}
|
|
|
|
} else {
|
|
pfxlog.Logger().WithFields(map[string]interface{}{
|
|
"routerId": r.Id,
|
|
"channel": r.Control.GetChannel().LogicalName(),
|
|
}).Info("no on connect settings to send")
|
|
}
|
|
}
|
|
|
|
type OnConnectCtrlAddressesUpdateHandler struct {
|
|
ctrlAddress string
|
|
raft *raft.Controller
|
|
}
|
|
|
|
func NewOnConnectCtrlAddressesUpdateHandler(ctrlAddress string, raft *raft.Controller) *OnConnectCtrlAddressesUpdateHandler {
|
|
return &OnConnectCtrlAddressesUpdateHandler{
|
|
ctrlAddress: ctrlAddress,
|
|
raft: raft,
|
|
}
|
|
}
|
|
|
|
func (o *OnConnectCtrlAddressesUpdateHandler) RouterDisconnected(r *model.Router) {
|
|
//do nothing, satisfy interface
|
|
}
|
|
|
|
func (o OnConnectCtrlAddressesUpdateHandler) RouterConnected(r *model.Router) {
|
|
index, data, controllers := o.raft.CtrlAddresses()
|
|
log := pfxlog.Logger().WithFields(map[string]interface{}{
|
|
"routerId": r.Id,
|
|
"channel": r.Control.GetChannel().LogicalName(),
|
|
"addresses": data,
|
|
"index": index,
|
|
})
|
|
|
|
if len(data) == 0 {
|
|
log.Error("no addresses to send")
|
|
return
|
|
}
|
|
|
|
log.Info("router connected, syncing ctrl addresses")
|
|
|
|
updMsg := &ctrl_pb.UpdateCtrlAddresses{
|
|
IsLeader: o.raft.IsLeader(),
|
|
Addresses: data,
|
|
Index: index,
|
|
Controllers: controllers,
|
|
}
|
|
|
|
if err := protobufs.MarshalTyped(updMsg).Send(r.Control.GetDefaultSender()); err != nil {
|
|
log.WithError(err).Error("error sending UpdateCtrlAddresses on router connect")
|
|
}
|
|
}
|